Merge pull request #2757 from AI-Hypercomputer:aireen/grain_mix2

PiperOrigin-RevId: 837310083

Author: Google-ML-Automation

dependencies/requirements/generated_requirements/cuda12-requirements.txt

@@ -15,20 +15,20 @@ array-record>=0.8.3
 astroid>=4.0.2
 astunparse>=1.6.3
 attrs>=25.4.0
-auditwheel>=6.4.2
+auditwheel>=6.5.0
 black>=24.10.0
 blobfile>=3.1.0
 build>=1.3.0
 cachetools>=6.2.2
-certifi>=2025.10.5
+certifi>=2025.11.12
 cfgv>=3.5.0
 charset-normalizer>=3.4.4
 cheroot>=11.1.2
 chex>=0.1.91
 click>=8.3.1
 cloud-accelerator-diagnostics>=0.1.1
 cloud-tpu-diagnostics>=0.1.5
-cloudpickle>=3.1.1
+cloudpickle>=3.1.2
 clu>=0.0.12
 colorama>=0.4.6
 contourpy>=1.3.3
@@ -45,16 +45,16 @@ einops>=0.8.1
 einshape>=1.0
 etils>=1.13.0
 evaluate>=0.4.6
-execnet>=2.1.1
-fastapi>=0.121.3
+execnet>=2.1.2
+fastapi>=0.122.0
 filelock>=3.20.0
 flatbuffers>=25.9.23
-flax>=0.12.0
+flax>=0.12.1
 fonttools>=4.60.1
 frozenlist>=1.8.0
-fsspec>=2025.9.0
+fsspec>=2025.10.0
 gast>=0.6.0
-gcsfs>=2025.9.0
+gcsfs>=2025.10.0
 google-api-core>=2.28.1
 google-api-python-client>=2.187.0
 google-auth-httplib2>=0.2.1
@@ -75,10 +75,10 @@ google-jetstream @ https://github.com/AI-Hypercomputer/JetStream/archive/29329e8
 google-pasta>=0.2.0
 google-resumable-media>=2.8.0
 googleapis-common-protos>=1.72.0
-grain>=0.2.14
+grain>=0.2.15
 grpc-google-iam-v1>=0.14.3
 grpcio-status>=1.71.2
-grpcio>=1.75.1
+grpcio>=1.76.0
 gviz-api>=1.10.0
 h11>=0.16.0
 h5py>=3.15.1
@@ -95,7 +95,7 @@ immutabledict>=4.2.2
 importlab>=0.8.1
 importlib-metadata>=8.7.0
 importlib-resources>=6.5.2
-iniconfig>=2.1.0
+iniconfig>=2.3.0
 isort>=7.0.0
 jaraco-functools>=4.3.0
 jax-cuda12-pjrt>=0.8.1 ; sys_platform == 'linux'
@@ -106,31 +106,31 @@ jaxtyping>=0.3.3
 jinja2>=3.1.6
 joblib>=1.5.2
 jsonlines>=4.0.0
-keras>=3.11.3
+keras>=3.12.0
 kiwisolver>=1.4.9
 libclang>=18.1.1
 libcst>=1.8.6
 lxml>=6.0.2
 markdown-it-py>=4.0.0
-markdown>=3.9
+markdown>=3.10
 markupsafe>=3.0.3
 matplotlib>=3.10.7
 mccabe>=0.7.0
 mdurl>=0.1.2
 ml-collections>=1.1.0
-ml-dtypes>=0.5.3
+ml-dtypes>=0.5.4
 ml-goodput-measurement>=0.0.15
 mlperf-logging @ https://github.com/mlcommons/logging/archive/38ab22670527888c8eb7825a4ece176fcc36a95d.zip
 more-itertools>=10.8.0
 mpmath>=1.3.0
 msgpack>=1.1.2
-msgspec>=0.19.0
+msgspec>=0.20.0
 multidict>=6.7.0
 multiprocess>=0.70.18
 mypy-extensions>=1.1.0
 namex>=0.1.0
 nest-asyncio>=1.6.0
-networkx>=3.5
+networkx>=3.6
 ninja>=1.13.0
 nltk>=3.9.2
 nodeenv>=1.9.1
@@ -141,19 +141,19 @@ nvidia-cuda-cupti-cu12>=12.9.79 ; sys_platform == 'linux'
 nvidia-cuda-nvcc-cu12>=12.9.86 ; sys_platform == 'linux'
 nvidia-cuda-nvrtc-cu12>=12.9.86 ; sys_platform == 'linux'
 nvidia-cuda-runtime-cu12>=12.9.79 ; sys_platform == 'linux'
-nvidia-cudnn-cu12>=9.14.0.64 ; sys_platform == 'linux'
+nvidia-cudnn-cu12>=9.16.0.29 ; sys_platform == 'linux'
 nvidia-cufft-cu12>=11.4.1.4 ; sys_platform == 'linux'
 nvidia-cusolver-cu12>=11.7.5.82 ; sys_platform == 'linux'
 nvidia-cusparse-cu12>=12.5.10.65 ; sys_platform == 'linux'
-nvidia-nccl-cu12>=2.28.3 ; sys_platform == 'linux'
+nvidia-nccl-cu12>=2.28.9 ; sys_platform == 'linux'
 nvidia-nvjitlink-cu12>=12.9.86 ; sys_platform == 'linux'
 nvidia-nvshmem-cu12>=3.4.5 ; sys_platform == 'linux'
 oauthlib>=3.3.1
 omegaconf>=2.3.0
 opentelemetry-api>=1.38.0
 opt-einsum>=3.4.0
 optax>=0.2.6
-optree>=0.17.0
+optree>=0.18.0
 optype>=0.14.0
 orbax-checkpoint>=0.11.28
 packaging>=25.0
@@ -165,20 +165,20 @@ pillow>=12.0.0
 platformdirs>=4.5.0
 pluggy>=1.6.0
 portpicker>=1.6.0
-pre-commit>=4.4.0
+pre-commit>=4.5.0
 prometheus-client>=0.23.1
 promise>=2.3
 propcache>=0.4.1
 proto-plus>=1.26.1
 protobuf>=5.29.5
-psutil>=7.1.0
+psutil>=7.1.3
 pyarrow>=22.0.0
 pyasn1-modules>=0.4.2
 pyasn1>=0.6.1
 pycnite>=2024.7.31
 pycryptodomex>=3.23.0
 pydantic-core>=2.41.5
-pydantic>=2.12.4
+pydantic>=2.12.5
 pydot>=4.0.1
 pyelftools>=0.32
 pyglove>=0.4.5
@@ -200,8 +200,8 @@ requests>=2.32.5
 rich>=14.2.0
 rsa>=4.9.1
 safetensors>=0.7.0
-scipy-stubs>=1.16.2.4
-scipy>=1.16.2
+scipy-stubs>=1.16.3.0
+scipy>=1.16.3
 sentencepiece>=0.2.1
 seqio>=0.0.20
 setuptools>=80.9.0
@@ -224,10 +224,10 @@ tensorflow-datasets>=4.9.9
 tensorflow-metadata>=1.17.2
 tensorflow-text>=2.19.0
 tensorflow>=2.19.1
-tensorstore>=0.1.78
-termcolor>=3.1.0
+tensorstore>=0.1.79
+termcolor>=3.2.0
 tiktoken>=0.12.0
-tokamax>=0.0.6
+tokamax>=0.0.8
 tokenizers>=0.22.1
 toml>=0.10.2
 tomlkit>=0.13.3
@@ -236,7 +236,7 @@ tqdm>=4.67.1
 transformer-engine-cu12>=2.9.0
 transformer-engine-jax>=2.9.0
 transformer-engine>=2.9.0
-transformers>=4.57.1
+transformers>=4.57.3
 treescope>=0.1.10
 typeguard>=2.13.3
 typing-extensions>=4.15.0

dependencies/requirements/generated_requirements/tpu-requirements.txt

@@ -15,24 +15,24 @@ array-record>=0.8.3
 astroid>=4.0.2
 astunparse>=1.6.3
 attrs>=25.4.0
-auditwheel>=6.4.2
+auditwheel>=6.5.0
 black>=24.10.0
 blobfile>=3.1.0
 build>=1.3.0
 cachetools>=6.2.2
-certifi>=2025.10.5
-cfgv>=3.4.0
+certifi>=2025.11.12
+cfgv>=3.5.0
 charset-normalizer>=3.4.4
 cheroot>=11.1.2
 chex>=0.1.91
 click>=8.3.1
 cloud-accelerator-diagnostics>=0.1.1
 cloud-tpu-diagnostics>=0.1.5
-cloudpickle>=3.1.1
+cloudpickle>=3.1.2
 clu>=0.0.12
 colorama>=0.4.6
 contourpy>=1.3.3
-coverage>=7.11.3
+coverage>=7.12.0
 cycler>=0.12.1
 datasets>=4.4.1
 decorator>=5.2.1
@@ -45,22 +45,22 @@ einops>=0.8.1
 einshape>=1.0
 etils>=1.13.0
 evaluate>=0.4.6
-execnet>=2.1.1
-fastapi>=0.121.2
+execnet>=2.1.2
+fastapi>=0.122.0
 filelock>=3.20.0
 flatbuffers>=25.9.23
-flax>=0.12.0
+flax>=0.12.1
 fonttools>=4.60.1
 frozenlist>=1.8.0
-fsspec>=2025.9.0
+fsspec>=2025.10.0
 gast>=0.6.0
-gcsfs>=2025.9.0
+gcsfs>=2025.10.0
 google-api-core>=2.28.1
 google-api-python-client>=2.187.0
 google-auth-httplib2>=0.2.1
 google-auth-oauthlib>=1.2.2
 google-auth>=2.43.0
-google-cloud-aiplatform>=1.127.0
+google-cloud-aiplatform>=1.128.0
 google-cloud-appengine-logging>=1.7.0
 google-cloud-audit-log>=0.4.0
 google-cloud-bigquery>=3.38.0
@@ -70,16 +70,16 @@ google-cloud-monitoring>=2.28.0
 google-cloud-resource-manager>=1.15.0
 google-cloud-storage>=3.6.0
 google-crc32c>=1.7.1
-google-genai>=1.50.1
+google-genai>=1.52.0
 google-jetstream @ https://github.com/AI-Hypercomputer/JetStream/archive/29329e8e73820993f77cfc8efe34eb2a73f5de98.zip
 google-pasta>=0.2.0
 google-resumable-media>=2.8.0
 google-tunix>=0.1.3
 googleapis-common-protos>=1.72.0
-grain>=0.2.14
+grain>=0.2.15
 grpc-google-iam-v1>=0.14.3
 grpcio-status>=1.71.2
-grpcio>=1.75.1
+grpcio>=1.76.0
 gviz-api>=1.10.0
 h11>=0.16.0
 h5py>=3.15.1
@@ -97,43 +97,43 @@ immutabledict>=4.2.2
 importlab>=0.8.1
 importlib-metadata>=8.7.0
 importlib-resources>=6.5.2
-iniconfig>=2.1.0
+iniconfig>=2.3.0
 isort>=7.0.0
 jaraco-functools>=4.3.0
-jax>=0.8.0
-jaxlib>=0.8.0
+jax>=0.8.1
+jaxlib>=0.8.1
 jaxtyping>=0.3.3
 jinja2>=3.1.6
 joblib>=1.5.2
 jsonlines>=4.0.0
 kagglehub>=0.3.13
-keras>=3.11.3
+keras>=3.12.0
 kiwisolver>=1.4.9
 libclang>=18.1.1
 libcst>=1.8.6
-libtpu>=0.0.24 ; platform_machine == 'x86_64' and sys_platform == 'linux'
+libtpu>=0.0.30 ; platform_machine == 'x86_64' and sys_platform == 'linux'
 llvmlite>=0.45.1
 lxml>=6.0.2
 markdown-it-py>=4.0.0
-markdown>=3.9
+markdown>=3.10
 markupsafe>=3.0.3
 matplotlib>=3.10.7
 mccabe>=0.7.0
 mdurl>=0.1.2
 ml-collections>=1.1.0
-ml-dtypes>=0.5.3
+ml-dtypes>=0.5.4
 ml-goodput-measurement>=0.0.15
 mlperf-logging @ https://github.com/mlcommons/logging/archive/38ab22670527888c8eb7825a4ece176fcc36a95d.zip
 more-itertools>=10.8.0
 mpmath>=1.3.0
 msgpack>=1.1.2
-msgspec>=0.19.0
+msgspec>=0.20.0
 multidict>=6.7.0
 multiprocess>=0.70.18
 mypy-extensions>=1.1.0
 namex>=0.1.0
 nest-asyncio>=1.6.0
-networkx>=3.5
+networkx>=3.6
 ninja>=1.13.0
 nltk>=3.9.2
 nodeenv>=1.9.1
@@ -145,7 +145,7 @@ omegaconf>=2.3.0
 opentelemetry-api>=1.38.0
 opt-einsum>=3.4.0
 optax>=0.2.6
-optree>=0.17.0
+optree>=0.18.0
 optype>=0.14.0
 orbax-checkpoint>=0.11.28
 packaging>=25.0
@@ -157,20 +157,20 @@ pillow>=12.0.0
 platformdirs>=4.5.0
 pluggy>=1.6.0
 portpicker>=1.6.0
-pre-commit>=4.4.0
+pre-commit>=4.5.0
 prometheus-client>=0.23.1
 promise>=2.3
 propcache>=0.4.1
 proto-plus>=1.26.1
 protobuf>=5.29.5
-psutil>=7.1.0
+psutil>=7.1.3
 pyarrow>=22.0.0
 pyasn1-modules>=0.4.2
 pyasn1>=0.6.1
 pycnite>=2024.7.31
 pycryptodomex>=3.23.0
 pydantic-core>=2.41.5
-pydantic>=2.12.4
+pydantic>=2.12.5
 pydot>=4.0.1
 pyelftools>=0.32
 pyglove>=0.4.5
@@ -192,9 +192,9 @@ requests-oauthlib>=2.0.0
 requests>=2.32.5
 rich>=14.2.0
 rsa>=4.9.1
-safetensors>=0.6.2
-scipy-stubs>=1.16.2.4
-scipy>=1.16.2
+safetensors>=0.7.0
+scipy-stubs>=1.16.3.0
+scipy>=1.16.3
 sentencepiece>=0.2.1
 seqio>=0.0.20
 setuptools>=80.9.0
@@ -205,7 +205,7 @@ simplejson>=3.20.2
 six>=1.17.0
 sniffio>=1.3.1
 sortedcontainers>=2.4.0
-starlette>=0.49.3
+starlette>=0.50.0
 sympy>=1.14.0
 tabulate>=0.9.0
 tenacity>=9.1.2
@@ -217,16 +217,16 @@ tensorflow-datasets>=4.9.9
 tensorflow-metadata>=1.17.2
 tensorflow-text>=2.19.0
 tensorflow>=2.19.1
-tensorstore>=0.1.78
-termcolor>=3.1.0
+tensorstore>=0.1.79
+termcolor>=3.2.0
 tiktoken>=0.12.0
-tokamax>=0.0.5
+tokamax>=0.0.8
 tokenizers>=0.22.1
 toml>=0.10.2
 tomlkit>=0.13.3
 toolz>=1.1.0
 tqdm>=4.67.1
-transformers>=4.57.1
+transformers>=4.57.3
 treescope>=0.1.10
 typeguard>=2.13.3
 typing-extensions>=4.15.0