[MRG] Temporary limit on jax version (#608)

rflamary · web-flow · commit 737b20db2f8d · 2024-02-29T16:40:44.000+01:00
* Update requirements.txt

Limit the max version of jax to allow for old config because of pymanopt using deprecated version

* Update requirements.txt
diff --git a/requirements.txt b/requirements.txt
@@ -6,11 +6,11 @@ pymanopt
 cvxopt
 scikit-learn
 torch
-jax
-jaxlib
+jax<=0.4.24
+jaxlib<=0.4.24
 tensorflow
 pytest
 torch_geometric
 cvxpy
 geomloss
-pykeops
+pykeops
