diff --git a/pyproject.toml b/pyproject.toml
index 64bfc4d..7f300a8 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -31,7 +31,7 @@ dependencies = [
     "h5py<=3.13.0",
     "iminuit<=2.30.1",
     "jax>=0.4.28; python_version < '3.10'",
-    "jax>=0.4.28,<=0.5.0; python_version >= '3.10'",
+    "jax>=0.4.28,<=0.5.1; python_version >= '3.10'",
     "jaxns==2.6.7",
     "matplotlib",
     "nautilus-sampler==1.0.5",