diff --git a/requirements.txt b/requirements.txt index dd810d0..8df4303 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,6 @@ pandas~=2.2 jax==0.4.30 -jaxlib==0.4.30 +jaxlib==0.5.1 jaxtyping==0.2.36 chex==0.1.87 optax==0.2.4