Require jaxlib 0.4.33 or newer

FlorianPfaff · Oct 30, 2024 · f3e3a4c · f3e3a4c
1 parent dba44db
commit f3e3a4c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -20,13 +20,13 @@ shapely = ">=2.0"
 [tool.poetry.extras]
 healpy_support = ["healpy"]
 pytorch_support = ["torch"]
-jax_support = ["jax", "jaxlib", "autograd"]
+jax_support = ["jax>=0.4.33", "jaxlib>=0.4.33", "autograd"]
 
 [tool.poetry.group.dev.dependencies]
 healpy = "*"
 torch = "*"
-jax = "*"
-jaxlib = "*"
+jax = ">=0.4.33"
+jaxlib = ">=0.4.33"
 autograd = "*"
 autopep8 = "^2.3.1"
 pytest = "*"