[CI] Remove jax upper bounds for testing

matthewfeickert · matthewfeickert · commit 6ecfbec52d24 · 2024-06-24T11:43:08.000-05:00
diff --git a/requirements_all.txt b/requirements_all.txt
@@ -6,8 +6,8 @@ pymanopt
 cvxopt
 scikit-learn
 torch
-jax<=0.4.24
-jaxlib<=0.4.24
+jax
+jaxlib
 tensorflow
 pytest
 torch_geometric
