diff --git a/pyproject.toml b/pyproject.toml
index 6e3d6fd..154db97 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -24,11 +24,11 @@ classifiers = [
 ]
 dependencies = [
 "jaxtyping>=0.2.0",
-"jax>=0.4.30",
+"jax>=0.4.35",
 ]
 
 [project.optional-dependencies]
-test = ["pytest>=8.0.0" , "jax>=0.4.30"]
+test = ["pytest>=8.0.0" , "jax>=0.4.35"]
 
 [tool.scikit-build]
 minimum-version = "0.8"