diff --git a/Dockerfile b/Dockerfile
index fa19e1ed39a5d4e92fa798edcd3ad15528a4f8d8..0445d2810bbfeaeb0968b0baeb5dba4ba01efb6f 100644
--- a/Dockerfile
+++ b/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
diff --git a/pyproject.toml b/pyproject.toml
index ca4c12b859dc838fcf6e8a9b3e0ba4271bd0bfee..76dd116d3fe5e2295a6f7541f4368aedf99eaa4d 100644
--- a/pyproject.toml
+++ b/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",
 ]