diff --git a/experimental/jax/requirements.txt b/experimental/jax/requirements.txt
index 72ba566c..68f9dd03 100644
--- a/experimental/jax/requirements.txt
+++ b/experimental/jax/requirements.txt
@@ -2,7 +2,7 @@
 --find-links https://storage.googleapis.com/jax-releases/libtpu_releases.html
 
 absl-py
-torch==2.3.0+cpu
+torch==2.6.0
 torchvision==0.18.0+cpu
 jax[tpu]==0.4.33
 huggingface_hub[cli]