diff --git a/requirements_jax.txt b/requirements_jax.txt index 6781de008..ab4988bec 100644 --- a/requirements_jax.txt +++ b/requirements_jax.txt @@ -1,2 +1,2 @@ -jaxlib==0.4.8 +jaxlib==0.4.13 jax==0.4.8