From 48e71c04ecfddb85db80a9d424a95a91f0ac6358 Mon Sep 17 00:00:00 2001
From: Gabriel Gerlero <gerlero@users.noreply.github.com>
Date: Mon, 26 Jan 2026 09:22:23 -0300
Subject: [PATCH] Update jax version constraint in pyproject.toml

---
 pyproject.toml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index 190a47e..7ce8eea 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -29,7 +29,7 @@ dependencies = [
     "diffrax>=0.7,<0.8",
     "equinox>=0.12.1,<0.14",
     "interpax>=0.3.8,<0.4",
-    "jax>=0.5,<0.9",
+    "jax>=0.5,<0.10",
     "numpy>=1,<3",
     "optax>=0.2.4,<0.3",
     "optimistix>=0.0.10,<0.1",