diff --git a/pyproject.toml b/pyproject.toml
index ad1def45..639038e5 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -26,7 +26,7 @@ dependencies = [
     "qnm==0.4.4",
     "seaborn~=0.13",
     "numpyro==0.18",
-    "jax>=0.4.25,<0.6.1",
+    "jax>=0.4.25,<0.6.2",
     "parse>=1.20",
     "jaxlib==0.4.38; sys_platform == 'darwin' and platform_machine == 'x86_64'",
 ]