fix to jax 0.4 until we can test 0.5

FLAIROx · Jan 29, 2025 · 3ffa5b8 · 3ffa5b8
1 parent 5ab338d
commit 3ffa5b8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -31,7 +31,7 @@ classifiers = [
     "License :: OSI Approved :: Apache Software License",
 ]
 dependencies = [
-  "jax",
+  "jax<=0.4.38",
   "jaxlib",
   "flax",
   "safetensors",