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