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