Revert "Upgrade jax version to 0.8.0" (vllm-project#1094)

vanbasten23 · AahilA · commit 6b0cd107ae4f · 2025-11-14T19:12:54.000Z
diff --git a/requirements.txt b/requirements.txt
@@ -5,8 +5,9 @@ pytest-mock
 absl-py
 numpy
 google-cloud-storage
-jax[tpu]==0.8.0
-jaxlib==0.8.0
+jax==0.7.2
+jaxlib==0.7.2
+libtpu==0.0.23
 jaxtyping
 flax==0.11.1
 torchax==0.0.7
