Pin also jaxlib

ami-iit · Nov 9, 2023 · 3b995fb · 3b995fb
1 parent 1403014
commit 3b995fb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.cfg b/setup.cfg
@@ -54,7 +54,7 @@ python_requires = >=3.10
 install_requires =
     coloredlogs
     jax >= 0.4.1, <0.4.11
-    jaxlib
+    jaxlib < 0.4.11
     jaxlie
     jax_dataclasses >= 1.4.0
     pptree