updating readme with new results

2024-06-03 18:03:38 +00:00
parent 7d4c2b8321
commit 84e8c931d6
5 changed files with 18 additions and 7 deletions
@@ -8,7 +8,7 @@ dependencies = [
    "einshape>=1.0.0",
    "paxml>=1.4.0",
    "praxis>=1.4.0",
-    "jax==0.4.26",
+    "jax>=0.4.26",
    "numpy>=1.26.4",
    "pandas>=2.1.4",
 ]