diff --git a/setup.py b/setup.py
index 3a02420e4..b93a735a0 100644
--- a/setup.py
+++ b/setup.py
@@ -9,7 +9,7 @@
 # `pyproject.toml`, once optional dependencies defined there can reference one another
 extras_require = dict()
 extras_require["jax"] = [
-    "jax[cpu]<0.3.5; platform_system!='Windows'",
+    "jax[cpu]<0.3.6; platform_system!='Windows'",
 ]
 extras_require["zoo"] = [
     "tqdm>=4.0",