diff --git a/setup.py b/setup.py
index 928969f13..33c3cf07a 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.4.20; platform_system!='Windows'",
+    "jax[cpu]<0.4.24; platform_system!='Windows'",
 ]
 extras_require["zoo"] = [
     "tqdm>=4.0",