diff --git a/requirements.txt b/requirements.txt index 0b3137c..5eec01f 100644 --- 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 jaxlib>=0.4.13 -grgrjax>=0.4.5 +grgrjax>=0.4.6 scipy pyyaml pytest diff --git a/setup.py b/setup.py index 9c89113..753c3bf 100644 --- a/setup.py +++ b/setup.py @@ -36,7 +36,7 @@ install_requires=[ "jax>=0.4.13", "jaxlib>=0.4.13", - "grgrjax>=0.4.5", + "grgrjax>=0.4.6", "pyyaml", "scipy", ],