Merge pull request #2538 from AI-Hypercomputer:mohit/fix_docker

PiperOrigin-RevId: 822796389

Author: Google-ML-Automation

generated_requirements/tpu-requirements.txt

@@ -60,7 +60,7 @@ google-auth-httplib2>=0.2.0
 google-auth-oauthlib>=1.2.2
 google-auth>=2.41.1
 google-benchmark>=1.9.4
-google-cloud-aiplatform>=1.121.0
+google-cloud-aiplatform>=1.122.0
 google-cloud-appengine-logging>=1.7.0
 google-cloud-audit-log>=0.4.0
 google-cloud-bigquery>=3.38.0
@@ -70,8 +70,8 @@ google-cloud-monitoring>=2.28.0
 google-cloud-resource-manager>=1.15.0
 google-cloud-storage>=2.19.0
 google-crc32c>=1.7.1
-google-genai>=1.45.0
-google-jetstream @ https://github.com/AI-Hypercomputer/JetStream/archive/daedc21c393f23449fb54ddc4f75fca34348ea9c.zip
+google-genai>=1.46.0
+google-jetstream @ https://github.com/AI-Hypercomputer/JetStream/archive/29329e8e73820993f77cfc8efe34eb2a73f5de98.zip
 google-pasta>=0.2.0
 google-resumable-media>=2.7.2
 google-tunix>=0.1.3
@@ -148,7 +148,7 @@ opt-einsum>=3.4.0
 optax>=0.2.6
 optree>=0.17.0
 optype>=0.14.0
-orbax-checkpoint>=0.11.25
+orbax-checkpoint>=0.11.26
 packaging>=25.0
 pandas>=2.3.3
 parameterized>=0.9.0