Update requirements.txt

rflamary · web-flow · commit 6f39c79a2756 · 2024-02-29T16:07:04.000+01:00
Limit the max version of jax to allow for old config because of pymanopt using deprecated version
diff --git a/requirements.txt b/requirements.txt
@@ -6,11 +6,11 @@ pymanopt
 cvxopt
 scikit-learn
 torch
-jax
+jax<=0.4.24
 jaxlib
 tensorflow
 pytest
 torch_geometric
 cvxpy
 geomloss
-pykeops
+pykeops
