diff --git a/pyproject.toml b/pyproject.toml index e62f534..b2f8efa 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -27,7 +27,7 @@ dependencies = [ "diffrax>=0.7.2,<0.8", "equinox>=0.13.8,<0.14", "interpax>=0.3.14,<0.4", - "jax>=0.10.2,<0.11", + "jax>=0.10.2,<0.12", "numpy>=2.4.6,<3", "optax>=0.2.8,<0.3", "optimistix>=0.1,<0.2",