diff --git a/python/sdist/pyproject.toml b/python/sdist/pyproject.toml
index c490b8331e..be2f4c97a1 100644
--- a/python/sdist/pyproject.toml
+++ b/python/sdist/pyproject.toml
@@ -87,7 +87,7 @@ examples = [
     "scipy",
 ]
 jax = [
-    "jax>=0.7.2",
+    "jax>=0.7.2,<0.8.2",
     "diffrax>=0.7.0",
     "jaxtyping>=0.2.34",
     "equinox>=0.13.2",