Skip to content

Commit d1e82b2

Browse files
SurbhiJainUSCGoogle-ML-Automation
authored andcommitted
Update libtpu to 0.0.38 in tpu-requirements.txt
PiperOrigin-RevId: 911434895
1 parent fb83eaa commit d1e82b2

7 files changed

Lines changed: 48 additions & 45 deletions

File tree

docs/development/update_dependencies.md

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -88,6 +88,7 @@ If you have made changes to TPU pre-training dependencies in `src/dependencies/r
8888
bash src/dependencies/scripts/generate_requirements.sh \
8989
--base-requirements src/dependencies/requirements/base_requirements/tpu-requirements.txt \
9090
--generated-requirements tpu-requirements.txt \
91+
--override-requirements src/dependencies/extra_deps/tpu_overrides.txt \
9192
--seed-commit <jax-build-commit-hash>
9293

9394
# Copy generated requirements to src/dependencies/requirements/generated_requirements
@@ -104,7 +105,7 @@ If you have made changes to the post-training dependencies in `src/dependencies/
104105
bash src/dependencies/scripts/generate_requirements.sh \
105106
--base-requirements src/dependencies/requirements/base_requirements/tpu-post-train-requirements.txt \
106107
--generated-requirements tpu-post-train-requirements.txt \
107-
--override-requirements src/dependencies/extra_deps/post_train_overrides.txt \
108+
--override-requirements src/dependencies/extra_deps/tpu_post_train_overrides.txt \
108109
--seed-commit <jax-build-commit-hash>
109110

110111
# Copy generated requirements to src/dependencies/requirements/generated_requirements
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
libtpu>=0.0.38

src/dependencies/extra_deps/post_train_overrides.txt renamed to src/dependencies/extra_deps/tpu_post_train_overrides.txt

File renamed without changes.

src/dependencies/requirements/generated_requirements/tpu-requirements.txt

Lines changed: 11 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ cloudpickle>=3.1.2
3030
clu>=0.0.12
3131
colorama>=0.4.6
3232
contourpy>=1.3.3
33-
cryptography>=47.0.0
33+
cryptography>=48.0.0
3434
cycler>=0.12.1
3535
dataclasses-json>=0.6.7
3636
datasets>=4.8.5
@@ -56,11 +56,11 @@ fsspec>=2026.2.0
5656
gast>=0.7.0
5757
gcsfs>=2026.2.0
5858
google-api-core>=2.30.3
59-
google-api-python-client>=2.194.0
60-
google-auth>=2.49.2
59+
google-api-python-client>=2.195.0
60+
google-auth>=2.50.0
6161
google-auth-httplib2>=0.3.1
6262
google-auth-oauthlib>=1.3.1
63-
google-cloud-aiplatform>=1.148.1
63+
google-cloud-aiplatform>=1.149.0
6464
google-cloud-appengine-logging>=1.9.0
6565
google-cloud-audit-log>=0.5.0
6666
google-cloud-bigquery>=3.41.0
@@ -72,22 +72,23 @@ google-cloud-resource-manager>=1.17.0
7272
google-cloud-storage>=3.10.1
7373
google-cloud-storage-control>=1.11.0
7474
google-crc32c>=1.8.0
75-
google-genai>=1.73.1
75+
google-genai>=1.75.0
7676
google-pasta>=0.2.0
7777
google-resumable-media>=2.8.2
7878
googleapis-common-protos>=1.74.0
7979
grain>=0.2.16
8080
grpc-google-iam-v1>=0.14.4
8181
grpcio>=1.78.0
8282
grpcio-status>=1.78.0
83+
grpcio-tools>=1.78.0
8384
gviz-api>=1.10.0
8485
h11>=0.16.0
8586
h5py>=3.14.0
8687
hf-xet>=1.4.3 ; platform_machine == 'AMD64' or platform_machine == 'aarch64' or platform_machine == 'amd64' or platform_machine == 'arm64' or platform_machine == 'x86_64'
8788
httpcore>=1.0.9
8889
httplib2>=0.31.2
8990
httpx>=0.28.1
90-
huggingface-hub>=1.12.0
91+
huggingface-hub>=1.13.0
9192
humanize>=4.15.0
9293
hypothesis>=6.142.1
9394
identify>=2.6.19
@@ -107,7 +108,7 @@ kiwisolver>=1.5.0
107108
latex2sympy2-extended>=1.11.0
108109
libclang>=18.1.1
109110
libcst>=1.8.6
110-
libtpu>=0.0.37 ; platform_machine == 'x86_64' and sys_platform == 'linux'
111+
libtpu>=0.0.38
111112
markdown>=3.10.2
112113
markdown-it-py>=4.0.0
113114
markupsafe>=3.0.3
@@ -140,7 +141,7 @@ opt-einsum>=3.4.0
140141
optax>=0.2.8
141142
optree>=0.19.0
142143
optype>=0.17.0
143-
orbax-checkpoint>=0.11.36
144+
orbax-checkpoint>=0.11.37
144145
orbax-export>=0.0.8
145146
packaging>=26.0
146147
pandas>=3.0.2
@@ -216,10 +217,10 @@ toml>=0.10.2
216217
tomlkit>=0.14.0
217218
toolz>=1.1.0
218219
tqdm>=4.67.3
219-
transformers>=5.6.2
220+
transformers>=5.8.0
220221
treescope>=0.1.10
221222
typeguard>=2.13.3
222-
typer>=0.25.0
223+
typer>=0.25.1
223224
typing-extensions>=4.15.0
224225
typing-inspect>=0.9.0
225226
typing-inspection>=0.4.2

src/dependencies/scripts/generate_requirements.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@
3030

3131
set -euo pipefail
3232

33-
SEED_ENV="$HOME/.local/bin/seed-env"
33+
SEED_ENV="seed-env"
3434

3535
# Defaults
3636
HARDWARE="tpu"

tests/utils/reference_hlo_llama3_8b.txt

Lines changed: 17 additions & 17 deletions
Large diffs are not rendered by default.

tests/utils/reference_hlo_qwen3_1.7b.txt

Lines changed: 16 additions & 16 deletions
Large diffs are not rendered by default.

0 commit comments

Comments
 (0)