From 73fb191496ace5d0155bd4c5fc439b5ec68f0810 Mon Sep 17 00:00:00 2001
From: Daniel Weindl <daniel.weindl@uni-bonn.de>
Date: Fri, 2 Jan 2026 08:55:18 +0100
Subject: [PATCH] Require jax<0.8.2

See #3113. For now, just require jax<0.8.2 to avoid test failures.
---
 python/sdist/pyproject.toml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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",