Remove jax version constraint as recent version works again

Dockerfile

@@ -12,7 +12,7 @@ RUN pip3 install --break-system-packages \
     # Packages needed for NIFTy
     scipy \
     # Optional nifty dependencies
-    matplotlib h5py astropy ducc0 "jax<=0.4.30" "jaxlib<=0.4.30" jaxbind mpi4py \
+    matplotlib h5py astropy ducc0 jax jaxlib jaxbind mpi4py \
     # Testing dependencies
     pytest pytest-cov pytest-xdist \
     # Documentation build dependencies

pyproject.toml

@@ -32,8 +32,8 @@ Homepage = "https://github.com/nifty-ppl/nifty"
 
 [project.optional-dependencies]
 re = [
-    "jax<=0.4.30",
-    "jaxlib<=0.4.30",
+    "jax",
+    "jaxlib",
     "jaxbind",
     "ducc0>=0.27.0",
 ]