diff --git a/pyproject.toml b/pyproject.toml
index aeab0f83..5cf70fd3 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -34,7 +34,7 @@ dev = [
     "coveralls>=4,<5",
     "ipykernel>=6,<7",
     "ipython>=8,<9",
-    "jax[cpu]>=0.4,<0.5",
+    "jax[cpu]>=0.4,<0.6",
     "jupyter>=1,<2",
     "mypy>=1,<2",
     "nbval>=0.11,<0.12",