Limit jax version.

YosefLab · Feb 12, 2024 · 4882539 · 4882539
1 parent a784a04
commit 4882539
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -30,10 +30,11 @@ dependencies = [
     "pandas<2.0.0,>=1.4.0",
     "bbknn>1.5.0",
     "celltypist>1.3.0",
-    "chex<0.1.3",
     "gdown>4.6.0",
     "h5py>3.7.0",
     "huggingface-hub>=0.11.1",
+    "jax==0.4.19", # Remove with scvi-tools 1.1
+    "jaxlib==0.4.19",
     "onclass>=1.3",
     "scanorama>1.7.2",
     "scanpy>1.9.0",