diff --git a/requirements.txt b/requirements.txt
index 8f421242b..b2f6e5f3f 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -5,8 +5,8 @@ pytest-mock
 absl-py
 numpy
 google-cloud-storage
-jax==0.7.2
-jaxlib==0.7.2
+jax==0.8.0
+jaxlib==0.8.0
 libtpu==0.0.23
 jaxtyping
 flax==0.11.1