grgrjax 0.4.7 compat

gboehl · gboehl · commit 3942e56e689a · 2025-03-20T13:28:57.000+01:00
diff --git a/requirements.txt b/requirements.txt
@@ -1,7 +1,7 @@
 # ubuntu seems to have issues with jax v0.4.6. Recheck when v0.4.7 is out
 jax>=0.4.13,<0.5.1
 jaxlib>=0.4.13,<0.5.1
-grgrjax>=0.4.6
+grgrjax>=0.4.7
 scipy
 pyyaml
 pytest
diff --git a/setup.py b/setup.py
@@ -36,7 +36,7 @@
     install_requires=[
         "jax>=0.4.13,<0.5.1",
         "jaxlib>=0.4.13,<0.5.1",
-        "grgrjax>=0.4.6",
+        "grgrjax>=0.4.7",
         "pyyaml",
         "scipy",
     ],
