update dependencies

flaport · Nov 6, 2024 · aa491c8 · aa491c8
1 parent 278d256
commit aa491c8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -27,9 +27,9 @@ dependencies = [
   "black",
   "fastcore",
   "flax>=0.8.2",
-  "jax",
+  "jax<0.4.35",
   "jaxlib<0.4.35",
-  "jaxtyping<0.4.35",
+  "jaxtyping",
   "klujax>=0.2.5",
   "matplotlib",
   "natsort",