restrict jax version (#313)

jfeser · web-flow · commit 6844bd338982 · 2025-07-24T18:05:36.000-04:00
diff --git a/pyproject.toml b/pyproject.toml
@@ -40,7 +40,7 @@ Source = "https://github.com/BasisResearch/effectful"
 [project.optional-dependencies]
 torch = ["torch"]
 pyro = ["pyro-ppl>=1.9.1"]
-jax = ["jax"]
+jax = ["jax<0.7"]
 numpyro = ["numpyro"]
 
 [dependency-groups]
