Merge pull request #107 from epignatelli/copilot/fix-104

Relax JAX constraint from jax<0.7 to jax

Author: epignatelli

navix/_version.py

@@ -18,5 +18,5 @@
 # under the License.
 
 
-__version__ = "0.7.3"
+__version__ = "0.7.4"
 __version_info__ = tuple(int(i) for i in __version__.split(".") if i.isdigit())

requirements.txt

@@ -9,7 +9,7 @@ setuptools_scm
 pytest
 # compute libraries
 pillow
-jax<0.7 # 0.7 breaks distrax
+jax
 flax
 rlax
 # experiments