diff --git a/requirements.txt b/requirements.txt index bc489a5..84fa830 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,2 +1,2 @@ -jax~=0.4.24 -jaxlib~=0.4.24 +jax~=0.4.25 +jaxlib~=0.4.25