diff --git a/pyproject.toml b/pyproject.toml
index 8cbb94c9..72c03595 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -26,7 +26,7 @@ dependencies = [
     "qnm==0.4.4",
     "seaborn~=0.13",
     "numpyro==0.19.0",
-    "jax>=0.4.25,<0.6.1",
+    "jax>=0.4.25,<0.7.1",
     "parse>=1.20",
     "jaxlib==0.4.38; sys_platform == 'darwin' and platform_machine == 'x86_64'",
     "tables~=3.10",