Closed sash-a closed 1 year ago
Unpin the jax version
It would be nice to always use the most recent jax
Change the requirements.txt, had to also switch to distrax master, see https://github.com/google-deepmind/distrax/issues/249
What?
Unpin the jax version
Why?
It would be nice to always use the most recent jax
How?
Change the requirements.txt, had to also switch to distrax master, see https://github.com/google-deepmind/distrax/issues/249