From 40db097fa44cde81fbb1568dd1b6ada7295a64e9 Mon Sep 17 00:00:00 2001 From: Dipannita Shaw Date: Thu, 5 Feb 2026 23:05:51 +0000 Subject: [PATCH 1/7] Update pinned to jax version 0.8.2 --- .../cuda12-requirements.txt | 196 ++++++++--------- .../tpu-requirements.txt | 197 +++++++++--------- 2 files changed, 204 insertions(+), 189 deletions(-) diff --git a/dependencies/requirements/generated_requirements/cuda12-requirements.txt b/dependencies/requirements/generated_requirements/cuda12-requirements.txt index 9879536ab1..aa112b005a 100644 --- a/dependencies/requirements/generated_requirements/cuda12-requirements.txt +++ b/dependencies/requirements/generated_requirements/cuda12-requirements.txt @@ -4,27 +4,26 @@ absl-py>=2.3.1 aiofiles>=25.1.0 aiohappyeyeballs>=2.6.1 -aiohttp>=3.13.2 +aiohttp>=3.13.3 aiosignal>=1.4.0 annotated-doc>=0.0.4 annotated-types>=0.7.0 antlr4-python3-runtime>=4.9.3 -anyio>=4.11.0 +anyio>=4.12.1 aqtp>=0.9.0 array-record>=0.8.3 -astroid>=4.0.2 +astroid>=4.0.3 astunparse>=1.6.3 attrs>=25.4.0 auditwheel>=6.5.0 -black>=24.10.0 +black>=25.12.0 blobfile>=3.1.0 build>=1.3.0 -cachetools>=6.2.2 certifi>=2025.11.12 +cffi>=2.0.0 ; platform_python_implementation != 'PyPy' 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 @@ -32,80 +31,84 @@ cloudpickle>=3.1.2 clu>=0.0.12 colorama>=0.4.6 contourpy>=1.3.3 -coverage>=7.12.0 +coverage>=7.13.3 +cryptography>=46.0.4 cycler>=0.12.1 -datasets>=4.4.1 +dataclasses-json>=0.6.7 +datasets>=4.5.0 decorator>=5.2.1 dill>=0.4.0 distlib>=0.4.0 +distro>=1.9.0 dm-tree>=0.1.9 docstring-parser>=0.17.0 drjax>=0.1.4 editdistance>=0.8.1 -einops>=0.8.1 +einops>=0.8.2 einshape>=1.0 etils>=1.13.0 evaluate>=0.4.6 execnet>=2.1.2 -fastapi>=0.122.0 -filelock>=3.20.0 +fastapi>=0.128.2 +filelock>=3.20.1 flatbuffers>=25.9.23 -flax>=0.12.1 -fonttools>=4.60.1 +flax>=0.12.3 +fonttools>=4.61.1 frozenlist>=1.8.0 fsspec>=2025.10.0 gast>=0.6.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.128.0 -google-cloud-appengine-logging>=1.7.0 +google-api-core>=2.29.0 +google-api-python-client>=2.189.0 +google-auth-httplib2>=0.3.0 +google-auth-oauthlib>=1.2.4 +google-auth>=2.48.0 +google-cloud-aiplatform>=1.136.0 +google-cloud-appengine-logging>=1.8.0 google-cloud-audit-log>=0.4.0 -google-cloud-bigquery>=3.38.0 +google-cloud-bigquery>=3.40.0 google-cloud-core>=2.5.0 -google-cloud-logging>=3.12.1 -google-cloud-mldiagnostics>=0.5.5 -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.52.0 +google-cloud-logging>=3.13.0 +google-cloud-mldiagnostics>=0.5.10 +google-cloud-monitoring>=2.29.1 +google-cloud-resource-manager>=1.16.0 +google-cloud-storage>=3.9.0 +google-crc32c>=1.8.0 +google-genai>=1.62.0 +google-jetstream @ https://github.com/AI-Hypercomputer/JetStream/archive/29329e8e73820993f77cfc8efe34eb2a73f5de98.zip google-pasta>=0.2.0 google-resumable-media>=2.8.0 googleapis-common-protos>=1.72.0 grain>=0.2.15 grpc-google-iam-v1>=0.14.3 -grpcio-status>=1.71.2 +grpcio-status>=1.76.0 grpcio>=1.76.0 gviz-api>=1.10.0 h11>=0.16.0 h5py>=3.15.1 -hf-xet>=1.2.0 ; platform_machine == 'aarch64' or platform_machine == 'amd64' or platform_machine == 'arm64' or platform_machine == 'x86_64' +hf-xet>=1.2.0 ; platform_machine == 'AMD64' or platform_machine == 'aarch64' or platform_machine == 'amd64' or platform_machine == 'arm64' or platform_machine == 'x86_64' httpcore>=1.0.9 -httplib2>=0.31.0 +httplib2>=0.31.2 httpx>=0.28.1 -huggingface-hub>=0.36.0 -humanize>=4.14.0 +huggingface-hub>=1.4.0 +humanize>=4.15.0 hypothesis>=6.142.1 -identify>=2.6.15 +identify>=2.6.16 idna>=3.11 immutabledict>=4.2.2 importlab>=0.8.1 -importlib-metadata>=8.7.0 +importlib-metadata>=8.7.1 importlib-resources>=6.5.2 iniconfig>=2.3.0 isort>=7.0.0 -jaraco-functools>=4.3.0 -jax-cuda12-pjrt>=0.8.1 ; sys_platform == 'linux' -jax-cuda12-plugin>=0.8.1 ; sys_platform == 'linux' -jax>=0.8.1 -jaxlib>=0.8.1 -jaxtyping>=0.3.3 +jaraco-functools>=4.4.0 +jax-cuda12-pjrt>=0.8.2 ; sys_platform == 'linux' +jax-cuda12-plugin>=0.8.2 ; sys_platform == 'linux' +jax>=0.8.3 +jaxlib>=0.8.2 +jaxtyping>=0.3.7 jinja2>=3.1.6 -joblib>=1.5.2 +joblib>=1.5.3 jsonlines>=4.0.0 keras>=3.12.0 kiwisolver>=1.4.9 @@ -115,26 +118,28 @@ lxml>=6.0.2 markdown-it-py>=4.0.0 markdown>=3.10 markupsafe>=3.0.3 -matplotlib>=3.10.7 +marshmallow>=3.26.2 +matplotlib>=3.10.8 mccabe>=0.7.0 mdurl>=0.1.2 ml-collections>=1.1.0 ml-dtypes>=0.5.4 -ml-goodput-measurement>=0.0.15 +ml-goodput-measurement>=0.0.16 +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.20.0 -multidict>=6.7.0 +multidict>=6.7.1 multiprocess>=0.70.18 mypy-extensions>=1.1.0 namex>=0.1.0 nest-asyncio>=1.6.0 -networkx>=3.6 +networkx>=3.6.1 ninja>=1.13.0 nltk>=3.9.2 -nodeenv>=1.9.1 -numpy-typing-compat>=20250818.2.0 +nodeenv>=1.10.0 +numpy-typing-compat>=20251206.2.0 numpy>=2.0.2 nvidia-cublas-cu12>=12.9.1.4 ; sys_platform == 'linux' nvidia-cuda-cupti-cu12>=12.9.79 ; sys_platform == 'linux' @@ -150,32 +155,34 @@ 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 +opentelemetry-api>=1.39.1 opt-einsum>=3.4.0 -optax>=0.2.6 +optax>=0.2.7 optree>=0.18.0 -optype>=0.14.0 -orbax-checkpoint>=0.11.28 +optype>=0.15.0 +orbax-checkpoint>=0.11.32 +orbax-export>=0.0.8 packaging>=25.0 -pandas>=2.3.3 +pandas>=3.0.0 parameterized>=0.9.0 -pathspec>=0.12.1 -pathwaysutils>=0.1.3 +pathspec>=1.0.4 +pathwaysutils>=0.1.4 pillow>=12.0.0 -platformdirs>=4.5.0 +platformdirs>=4.5.1 pluggy>=1.6.0 portpicker>=1.6.0 -pre-commit>=4.5.0 -prometheus-client>=0.23.1 +pre-commit>=4.5.1 +prometheus-client>=0.24.1 promise>=2.3 propcache>=0.4.1 -proto-plus>=1.26.1 -protobuf>=5.29.5 +proto-plus>=1.27.1 +protobuf>=6.33.5 psutil>=7.1.3 -pyarrow>=22.0.0 +pyarrow>=23.0.0 pyasn1-modules>=0.4.2 -pyasn1>=0.6.1 +pyasn1>=0.6.2 pycnite>=2024.7.31 +pycparser>=3.0 ; implementation_name != 'PyPy' and platform_python_implementation != 'PyPy' pycryptodomex>=3.23.0 pydantic-core>=2.41.5 pydantic>=2.12.5 @@ -183,31 +190,31 @@ pydot>=4.0.1 pyelftools>=0.32 pyglove>=0.4.5 pygments>=2.19.2 -pyink>=24.10.1 -pylint>=4.0.3 +pyink>=25.12.0 +pylint>=4.0.4 pyparsing>=3.2.5 pyproject-hooks>=1.2.0 pytest-xdist>=3.8.0 pytest>=8.4.2 python-dateutil>=2.9.0.post0 +pytokens>=0.4.1 pytype>=2024.10.11 -pytz>=2025.2 pyyaml>=6.0.3 -qwix>=0.1.4 -regex>=2025.11.3 +qwix>=0.1.5 +regex>=2026.1.15 requests-oauthlib>=2.0.0 requests>=2.32.5 rich>=14.2.0 rsa>=4.9.1 safetensors>=0.7.0 -scipy-stubs>=1.16.3.0 +scipy-stubs>=1.16.3.3 scipy>=1.16.3 sentencepiece>=0.2.1 seqio>=0.0.20 setuptools>=80.9.0 -shapely>=2.1.2 +shellingham>=1.5.4 shortuuid>=1.0.13 -simple-parsing>=0.1.7 +simple-parsing>=0.1.8 simplejson>=3.20.2 six>=1.17.0 sniffio>=1.3.1 @@ -215,44 +222,45 @@ sortedcontainers>=2.4.0 starlette>=0.50.0 sympy>=1.14.0 tabulate>=0.9.0 -tenacity>=9.1.2 +tenacity>=9.1.3 tensorboard-data-server>=0.7.2 tensorboard-plugin-profile>=2.13.0 -tensorboard>=2.19.0 +tensorboard>=2.20.0 tensorboardx>=2.6.4 tensorflow-datasets>=4.9.9 -tensorflow-metadata>=1.17.2 -tensorflow-text>=2.19.0 -tensorflow>=2.19.1 -tensorstore>=0.1.79 +tensorflow-metadata>=1.17.3 +tensorflow-text>=2.20.0 +tensorflow>=2.20.0 +tensorstore>=0.1.80 termcolor>=3.2.0 tiktoken>=0.12.0 -tokamax>=0.0.8 -tokenizers>=0.22.1 +tokamax>=0.0.10 +tokenizers>=0.22.2 toml>=0.10.2 -tomlkit>=0.13.3 -toolz>=1.1.0 -tqdm>=4.67.1 -transformer-engine-cu12>=2.9.0 -transformer-engine-jax>=2.9.0 -transformer-engine>=2.9.0 -transformers>=4.57.3 +tomlkit>=0.14.0 +tqdm>=4.67.3 +transformer-engine-cu12>=2.11.0 +transformer-engine-jax>=2.11.0 +transformer-engine>=2.11.0 +transformers>=5.1.0 treescope>=0.1.10 typeguard>=2.13.3 +typer-slim>=0.21.1 typing-extensions>=4.15.0 +typing-inspect>=0.9.0 typing-inspection>=0.4.2 -tzdata>=2025.2 +tzdata>=2025.3 ; sys_platform == 'emscripten' or sys_platform == 'win32' uritemplate>=4.2.0 -urllib3>=2.5.0 -uvicorn>=0.38.0 -virtualenv>=20.35.4 +urllib3>=2.6.2 +uvicorn>=0.40.0 +virtualenv>=20.36.1 wadler-lindig>=0.1.7 websockets>=15.0.1 -werkzeug>=3.1.3 -wheel>=0.45.1 -wrapt>=2.0.1 -xprof>=2.21.1 +werkzeug>=3.1.4 +wheel>=0.46.1 +wrapt>=2.1.1 +xprof>=2.21.5 xxhash>=3.6.0 yarl>=1.22.0 zipp>=3.23.0 -zstandard>=0.25.0 +zstandard>=0.25.0 \ No newline at end of file diff --git a/dependencies/requirements/generated_requirements/tpu-requirements.txt b/dependencies/requirements/generated_requirements/tpu-requirements.txt index 21f5668c98..c61aea7099 100644 --- a/dependencies/requirements/generated_requirements/tpu-requirements.txt +++ b/dependencies/requirements/generated_requirements/tpu-requirements.txt @@ -4,27 +4,26 @@ absl-py>=2.3.1 aiofiles>=25.1.0 aiohappyeyeballs>=2.6.1 -aiohttp>=3.13.2 +aiohttp>=3.13.3 aiosignal>=1.4.0 annotated-doc>=0.0.4 annotated-types>=0.7.0 antlr4-python3-runtime>=4.9.3 -anyio>=4.11.0 +anyio>=4.12.1 aqtp>=0.9.0 array-record>=0.8.3 -astroid>=4.0.2 +astroid>=4.0.3 astunparse>=1.6.3 attrs>=25.4.0 auditwheel>=6.5.0 -black>=24.10.0 +black>=25.12.0 blobfile>=3.1.0 build>=1.3.0 -cachetools>=6.2.2 certifi>=2025.11.12 +cffi>=2.0.0 ; platform_python_implementation != 'PyPy' 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 @@ -32,144 +31,151 @@ cloudpickle>=3.1.2 clu>=0.0.12 colorama>=0.4.6 contourpy>=1.3.3 -coverage>=7.12.0 +coverage>=7.13.3 +cryptography>=46.0.4 cycler>=0.12.1 -dacite>=1.9.2 -datasets>=4.4.1 +dataclasses-json>=0.6.7 +datasets>=4.5.0 decorator>=5.2.1 dill>=0.4.0 distlib>=0.4.0 +distro>=1.9.0 dm-tree>=0.1.9 docstring-parser>=0.17.0 drjax>=0.1.4 editdistance>=0.8.1 -einops>=0.8.1 +einops>=0.8.2 einshape>=1.0 etils>=1.13.0 evaluate>=0.4.6 execnet>=2.1.2 -fastapi>=0.122.0 -filelock>=3.20.0 +fastapi>=0.128.2 +filelock>=3.20.1 flatbuffers>=25.9.23 -flax>=0.12.1 -fonttools>=4.60.1 +flax>=0.12.3 +fonttools>=4.61.1 frozenlist>=1.8.0 fsspec>=2025.10.0 gast>=0.6.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.128.0 -google-cloud-appengine-logging>=1.7.0 +google-api-core>=2.29.0 +google-api-python-client>=2.189.0 +google-auth-httplib2>=0.3.0 +google-auth-oauthlib>=1.2.4 +google-auth>=2.48.0 +google-cloud-aiplatform>=1.136.0 +google-cloud-appengine-logging>=1.8.0 google-cloud-audit-log>=0.4.0 -google-cloud-bigquery>=3.38.0 +google-cloud-bigquery>=3.40.0 google-cloud-core>=2.5.0 -google-cloud-logging>=3.12.1 -google-cloud-mldiagnostics>=0.5.5 -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.52.0 +google-cloud-logging>=3.13.0 +google-cloud-mldiagnostics>=0.5.10 +google-cloud-monitoring>=2.29.1 +google-cloud-resource-manager>=1.16.0 +google-cloud-storage>=3.9.0 +google-crc32c>=1.8.0 +google-genai>=1.62.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.15 grpc-google-iam-v1>=0.14.3 -grpcio-status>=1.71.2 +grpcio-status>=1.76.0 grpcio>=1.76.0 -gspread>=6.2.1 gviz-api>=1.10.0 h11>=0.16.0 h5py>=3.15.1 hf-transfer>=0.1.9 -hf-xet>=1.2.0 ; platform_machine == 'aarch64' or platform_machine == 'amd64' or platform_machine == 'arm64' or platform_machine == 'x86_64' +hf-xet>=1.2.0 ; platform_machine == 'AMD64' or platform_machine == 'aarch64' or platform_machine == 'amd64' or platform_machine == 'arm64' or platform_machine == 'x86_64' httpcore>=1.0.9 -httplib2>=0.31.0 +httplib2>=0.31.2 httpx>=0.28.1 -huggingface-hub>=0.36.0 -humanize>=4.14.0 +huggingface-hub>=1.4.0 +humanize>=4.15.0 hypothesis>=6.142.1 -identify>=2.6.15 +identify>=2.6.16 idna>=3.11 immutabledict>=4.2.2 importlab>=0.8.1 -importlib-metadata>=8.7.0 +importlib-metadata>=8.7.1 importlib-resources>=6.5.2 iniconfig>=2.3.0 isort>=7.0.0 -jaraco-functools>=4.3.0 -jax>=0.8.1 -jaxlib>=0.8.1 -jaxtyping>=0.3.3 +jaraco-functools>=4.4.0 +jax>=0.8.3 +jaxlib>=0.8.2 +jaxtyping>=0.3.7 jinja2>=3.1.6 -joblib>=1.5.2 +joblib>=1.5.3 jsonlines>=4.0.0 -kagglehub>=0.3.13 +kagglehub>=0.4.2 +kagglesdk>=0.1.15 keras>=3.12.0 kiwisolver>=1.4.9 libclang>=18.1.1 libcst>=1.8.6 -libtpu>=0.0.30 ; platform_machine == 'x86_64' and sys_platform == 'linux' -llvmlite>=0.45.1 +libtpu>=0.0.32 ; platform_machine == 'x86_64' and sys_platform == 'linux' +llvmlite>=0.46.0 lxml>=6.0.2 markdown-it-py>=4.0.0 markdown>=3.10 markupsafe>=3.0.3 -matplotlib>=3.10.7 +marshmallow>=3.26.2 +matplotlib>=3.10.8 mccabe>=0.7.0 mdurl>=0.1.2 ml-collections>=1.1.0 ml-dtypes>=0.5.4 -ml-goodput-measurement>=0.0.15 +ml-goodput-measurement>=0.0.16 +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.20.0 -multidict>=6.7.0 +multidict>=6.7.1 multiprocess>=0.70.18 mypy-extensions>=1.1.0 namex>=0.1.0 nest-asyncio>=1.6.0 -networkx>=3.6 +networkx>=3.6.1 ninja>=1.13.0 nltk>=3.9.2 -nodeenv>=1.9.1 -numba>=0.62.1 -numpy-typing-compat>=20250818.2.0 +nodeenv>=1.10.0 +numba>=0.63.1 +numpy-typing-compat>=20251206.2.0 numpy>=2.0.2 oauthlib>=3.3.1 omegaconf>=2.3.0 -opentelemetry-api>=1.38.0 +opentelemetry-api>=1.39.1 opt-einsum>=3.4.0 -optax>=0.2.6 +optax>=0.2.7 optree>=0.18.0 -optype>=0.14.0 -orbax-checkpoint>=0.11.28 +optype>=0.15.0 +orbax-checkpoint>=0.11.32 +orbax-export>=0.0.8 packaging>=25.0 -pandas>=2.3.3 +pandas>=3.0.0 parameterized>=0.9.0 -pathspec>=0.12.1 -pathwaysutils>=0.1.3 +pathspec>=1.0.4 +pathwaysutils>=0.1.4 pillow>=12.0.0 -platformdirs>=4.5.0 +platformdirs>=4.5.1 pluggy>=1.6.0 portpicker>=1.6.0 -pre-commit>=4.5.0 -prometheus-client>=0.23.1 +pre-commit>=4.5.1 +prometheus-client>=0.24.1 promise>=2.3 propcache>=0.4.1 -proto-plus>=1.26.1 -protobuf>=5.29.5 +proto-plus>=1.27.1 +protobuf>=6.33.5 psutil>=7.1.3 -pyarrow>=22.0.0 +pyarrow>=23.0.0 pyasn1-modules>=0.4.2 -pyasn1>=0.6.1 +pyasn1>=0.6.2 pycnite>=2024.7.31 +pycparser>=3.0 ; implementation_name != 'PyPy' and platform_python_implementation != 'PyPy' pycryptodomex>=3.23.0 pydantic-core>=2.41.5 pydantic>=2.12.5 @@ -177,32 +183,32 @@ pydot>=4.0.1 pyelftools>=0.32 pyglove>=0.4.5 pygments>=2.19.2 -pyink>=24.10.1 -pylint>=4.0.3 +pyink>=25.12.0 +pylint>=4.0.4 pyparsing>=3.2.5 pyproject-hooks>=1.2.0 pytest-xdist>=3.8.0 pytest>=8.4.2 python-dateutil>=2.9.0.post0 python-dotenv>=1.2.1 +pytokens>=0.4.1 pytype>=2024.10.11 -pytz>=2025.2 pyyaml>=6.0.3 -qwix>=0.1.4 -regex>=2025.11.3 +qwix>=0.1.5 +regex>=2026.1.15 requests-oauthlib>=2.0.0 requests>=2.32.5 rich>=14.2.0 rsa>=4.9.1 safetensors>=0.7.0 -scipy-stubs>=1.16.3.0 +scipy-stubs>=1.16.3.3 scipy>=1.16.3 sentencepiece>=0.2.1 seqio>=0.0.20 setuptools>=80.9.0 -shapely>=2.1.2 +shellingham>=1.5.4 shortuuid>=1.0.13 -simple-parsing>=0.1.7 +simple-parsing>=0.1.8 simplejson>=3.20.2 six>=1.17.0 sniffio>=1.3.1 @@ -210,41 +216,42 @@ sortedcontainers>=2.4.0 starlette>=0.50.0 sympy>=1.14.0 tabulate>=0.9.0 -tenacity>=9.1.2 +tenacity>=9.1.3 tensorboard-data-server>=0.7.2 tensorboard-plugin-profile>=2.13.0 -tensorboard>=2.19.0 +tensorboard>=2.20.0 tensorboardx>=2.6.4 tensorflow-datasets>=4.9.9 -tensorflow-metadata>=1.17.2 -tensorflow-text>=2.19.0 -tensorflow>=2.19.1 -tensorstore>=0.1.79 +tensorflow-metadata>=1.17.3 +tensorflow-text>=2.20.0 +tensorflow>=2.20.0 +tensorstore>=0.1.80 termcolor>=3.2.0 tiktoken>=0.12.0 -tokamax>=0.0.8 -tokenizers>=0.22.1 +tokamax>=0.0.10 +tokenizers>=0.22.2 toml>=0.10.2 -tomlkit>=0.13.3 -toolz>=1.1.0 -tqdm>=4.67.1 -transformers>=4.57.3 +tomlkit>=0.14.0 +tqdm>=4.67.3 +transformers>=5.1.0 treescope>=0.1.10 typeguard>=2.13.3 +typer-slim>=0.21.1 typing-extensions>=4.15.0 +typing-inspect>=0.9.0 typing-inspection>=0.4.2 -tzdata>=2025.2 +tzdata>=2025.3 ; sys_platform == 'emscripten' or sys_platform == 'win32' uritemplate>=4.2.0 -urllib3>=2.5.0 -uvicorn>=0.38.0 -virtualenv>=20.35.4 +urllib3>=2.6.2 +uvicorn>=0.40.0 +virtualenv>=20.36.1 wadler-lindig>=0.1.7 websockets>=15.0.1 -werkzeug>=3.1.3 -wheel>=0.45.1 -wrapt>=2.0.1 -xprof>=2.21.1 +werkzeug>=3.1.4 +wheel>=0.46.1 +wrapt>=2.1.1 +xprof>=2.21.5 xxhash>=3.6.0 yarl>=1.22.0 zipp>=3.23.0 -zstandard>=0.25.0 +zstandard>=0.25.0 \ No newline at end of file From 75fc9bd6e5f0e87b44299cf5dae6a895cdc53dc9 Mon Sep 17 00:00:00 2001 From: Dipannita Shaw Date: Fri, 6 Feb 2026 05:19:25 +0000 Subject: [PATCH 2/7] Trigger CI re-run From 9a4dc5f4d17d32e9386115c7f728d2c78ffbebaf Mon Sep 17 00:00:00 2001 From: Dipannita Shaw Date: Fri, 6 Feb 2026 21:19:57 +0000 Subject: [PATCH 3/7] Pin some deps --- .../requirements/base_requirements/requirements.txt | 5 +++-- .../generated_requirements/cuda12-requirements.txt | 12 +++++------- .../generated_requirements/tpu-requirements.txt | 12 +++++------- 3 files changed, 13 insertions(+), 16 deletions(-) diff --git a/dependencies/requirements/base_requirements/requirements.txt b/dependencies/requirements/base_requirements/requirements.txt index c40252cfc1..bc82dc94d9 100644 --- a/dependencies/requirements/base_requirements/requirements.txt +++ b/dependencies/requirements/base_requirements/requirements.txt @@ -12,7 +12,8 @@ google-cloud-aiplatform google-cloud-mldiagnostics google-cloud-monitoring grain[parquet] -huggingface_hub +huggingface_hub==0.36.0 +hf-xet!=1.2.1 jax jaxlib jaxtyping @@ -40,7 +41,7 @@ tensorflow-text tensorflow tiktoken tokamax -transformers +transformers==4.57.6 qwix google-jetstream @ https://github.com/AI-Hypercomputer/JetStream/archive/29329e8e73820993f77cfc8efe34eb2a73f5de98.zip mlperf-logging @ https://github.com/mlcommons/logging/archive/38ab22670527888c8eb7825a4ece176fcc36a95d.zip diff --git a/dependencies/requirements/generated_requirements/cuda12-requirements.txt b/dependencies/requirements/generated_requirements/cuda12-requirements.txt index aa112b005a..5cca458493 100644 --- a/dependencies/requirements/generated_requirements/cuda12-requirements.txt +++ b/dependencies/requirements/generated_requirements/cuda12-requirements.txt @@ -49,7 +49,7 @@ einshape>=1.0 etils>=1.13.0 evaluate>=0.4.6 execnet>=2.1.2 -fastapi>=0.128.2 +fastapi>=0.128.3 filelock>=3.20.1 flatbuffers>=25.9.23 flax>=0.12.3 @@ -86,11 +86,11 @@ grpcio>=1.76.0 gviz-api>=1.10.0 h11>=0.16.0 h5py>=3.15.1 -hf-xet>=1.2.0 ; platform_machine == 'AMD64' or platform_machine == 'aarch64' or platform_machine == 'amd64' or platform_machine == 'arm64' or platform_machine == 'x86_64' +hf-xet>=1.2.0 httpcore>=1.0.9 httplib2>=0.31.2 httpx>=0.28.1 -huggingface-hub>=1.4.0 +huggingface-hub>=0.36.0 humanize>=4.15.0 hypothesis>=6.142.1 identify>=2.6.16 @@ -212,14 +212,13 @@ scipy>=1.16.3 sentencepiece>=0.2.1 seqio>=0.0.20 setuptools>=80.9.0 -shellingham>=1.5.4 shortuuid>=1.0.13 simple-parsing>=0.1.8 simplejson>=3.20.2 six>=1.17.0 sniffio>=1.3.1 sortedcontainers>=2.4.0 -starlette>=0.50.0 +starlette>=0.52.1 sympy>=1.14.0 tabulate>=0.9.0 tenacity>=9.1.3 @@ -242,10 +241,9 @@ tqdm>=4.67.3 transformer-engine-cu12>=2.11.0 transformer-engine-jax>=2.11.0 transformer-engine>=2.11.0 -transformers>=5.1.0 +transformers>=4.57.6 treescope>=0.1.10 typeguard>=2.13.3 -typer-slim>=0.21.1 typing-extensions>=4.15.0 typing-inspect>=0.9.0 typing-inspection>=0.4.2 diff --git a/dependencies/requirements/generated_requirements/tpu-requirements.txt b/dependencies/requirements/generated_requirements/tpu-requirements.txt index c61aea7099..19a05b80c5 100644 --- a/dependencies/requirements/generated_requirements/tpu-requirements.txt +++ b/dependencies/requirements/generated_requirements/tpu-requirements.txt @@ -49,7 +49,7 @@ einshape>=1.0 etils>=1.13.0 evaluate>=0.4.6 execnet>=2.1.2 -fastapi>=0.128.2 +fastapi>=0.128.3 filelock>=3.20.1 flatbuffers>=25.9.23 flax>=0.12.3 @@ -88,11 +88,11 @@ gviz-api>=1.10.0 h11>=0.16.0 h5py>=3.15.1 hf-transfer>=0.1.9 -hf-xet>=1.2.0 ; platform_machine == 'AMD64' or platform_machine == 'aarch64' or platform_machine == 'amd64' or platform_machine == 'arm64' or platform_machine == 'x86_64' +hf-xet>=1.2.0 httpcore>=1.0.9 httplib2>=0.31.2 httpx>=0.28.1 -huggingface-hub>=1.4.0 +huggingface-hub>=0.36.0 humanize>=4.15.0 hypothesis>=6.142.1 identify>=2.6.16 @@ -206,14 +206,13 @@ scipy>=1.16.3 sentencepiece>=0.2.1 seqio>=0.0.20 setuptools>=80.9.0 -shellingham>=1.5.4 shortuuid>=1.0.13 simple-parsing>=0.1.8 simplejson>=3.20.2 six>=1.17.0 sniffio>=1.3.1 sortedcontainers>=2.4.0 -starlette>=0.50.0 +starlette>=0.52.1 sympy>=1.14.0 tabulate>=0.9.0 tenacity>=9.1.3 @@ -233,10 +232,9 @@ tokenizers>=0.22.2 toml>=0.10.2 tomlkit>=0.14.0 tqdm>=4.67.3 -transformers>=5.1.0 +transformers>=4.57.6 treescope>=0.1.10 typeguard>=2.13.3 -typer-slim>=0.21.1 typing-extensions>=4.15.0 typing-inspect>=0.9.0 typing-inspection>=0.4.2 From b4f46f53e7a9ddec7d7ea4d99b34661dd3091f71 Mon Sep 17 00:00:00 2001 From: Dipannita Shaw Date: Fri, 6 Feb 2026 22:18:04 +0000 Subject: [PATCH 4/7] Pin some more deps --- .../requirements/base_requirements/requirements.txt | 6 +++--- .../generated_requirements/cuda12-requirements.txt | 6 +----- .../generated_requirements/tpu-requirements.txt | 8 ++------ 3 files changed, 6 insertions(+), 14 deletions(-) diff --git a/dependencies/requirements/base_requirements/requirements.txt b/dependencies/requirements/base_requirements/requirements.txt index bc82dc94d9..27941b0ca9 100644 --- a/dependencies/requirements/base_requirements/requirements.txt +++ b/dependencies/requirements/base_requirements/requirements.txt @@ -5,7 +5,7 @@ cloud-accelerator-diagnostics cloud-tpu-diagnostics datasets drjax -flax +flax!=0.12.3 gcsfs google-api-python-client google-cloud-aiplatform @@ -40,8 +40,8 @@ tensorflow-datasets tensorflow-text tensorflow tiktoken -tokamax -transformers==4.57.6 +tokamax!=0.1.0 +transformers qwix google-jetstream @ https://github.com/AI-Hypercomputer/JetStream/archive/29329e8e73820993f77cfc8efe34eb2a73f5de98.zip mlperf-logging @ https://github.com/mlcommons/logging/archive/38ab22670527888c8eb7825a4ece176fcc36a95d.zip diff --git a/dependencies/requirements/generated_requirements/cuda12-requirements.txt b/dependencies/requirements/generated_requirements/cuda12-requirements.txt index 5cca458493..7ccaf4ec31 100644 --- a/dependencies/requirements/generated_requirements/cuda12-requirements.txt +++ b/dependencies/requirements/generated_requirements/cuda12-requirements.txt @@ -34,7 +34,6 @@ contourpy>=1.3.3 coverage>=7.13.3 cryptography>=46.0.4 cycler>=0.12.1 -dataclasses-json>=0.6.7 datasets>=4.5.0 decorator>=5.2.1 dill>=0.4.0 @@ -52,7 +51,7 @@ execnet>=2.1.2 fastapi>=0.128.3 filelock>=3.20.1 flatbuffers>=25.9.23 -flax>=0.12.3 +flax>=0.12.2 fonttools>=4.61.1 frozenlist>=1.8.0 fsspec>=2025.10.0 @@ -118,7 +117,6 @@ lxml>=6.0.2 markdown-it-py>=4.0.0 markdown>=3.10 markupsafe>=3.0.3 -marshmallow>=3.26.2 matplotlib>=3.10.8 mccabe>=0.7.0 mdurl>=0.1.2 @@ -161,7 +159,6 @@ optax>=0.2.7 optree>=0.18.0 optype>=0.15.0 orbax-checkpoint>=0.11.32 -orbax-export>=0.0.8 packaging>=25.0 pandas>=3.0.0 parameterized>=0.9.0 @@ -245,7 +242,6 @@ transformers>=4.57.6 treescope>=0.1.10 typeguard>=2.13.3 typing-extensions>=4.15.0 -typing-inspect>=0.9.0 typing-inspection>=0.4.2 tzdata>=2025.3 ; sys_platform == 'emscripten' or sys_platform == 'win32' uritemplate>=4.2.0 diff --git a/dependencies/requirements/generated_requirements/tpu-requirements.txt b/dependencies/requirements/generated_requirements/tpu-requirements.txt index 19a05b80c5..6cba264033 100644 --- a/dependencies/requirements/generated_requirements/tpu-requirements.txt +++ b/dependencies/requirements/generated_requirements/tpu-requirements.txt @@ -34,7 +34,6 @@ contourpy>=1.3.3 coverage>=7.13.3 cryptography>=46.0.4 cycler>=0.12.1 -dataclasses-json>=0.6.7 datasets>=4.5.0 decorator>=5.2.1 dill>=0.4.0 @@ -52,7 +51,7 @@ execnet>=2.1.2 fastapi>=0.128.3 filelock>=3.20.1 flatbuffers>=25.9.23 -flax>=0.12.3 +flax>=0.12.2 fonttools>=4.61.1 frozenlist>=1.8.0 fsspec>=2025.10.0 @@ -110,7 +109,7 @@ jaxtyping>=0.3.7 jinja2>=3.1.6 joblib>=1.5.3 jsonlines>=4.0.0 -kagglehub>=0.4.2 +kagglehub>=0.4.3 kagglesdk>=0.1.15 keras>=3.12.0 kiwisolver>=1.4.9 @@ -122,7 +121,6 @@ lxml>=6.0.2 markdown-it-py>=4.0.0 markdown>=3.10 markupsafe>=3.0.3 -marshmallow>=3.26.2 matplotlib>=3.10.8 mccabe>=0.7.0 mdurl>=0.1.2 @@ -154,7 +152,6 @@ optax>=0.2.7 optree>=0.18.0 optype>=0.15.0 orbax-checkpoint>=0.11.32 -orbax-export>=0.0.8 packaging>=25.0 pandas>=3.0.0 parameterized>=0.9.0 @@ -236,7 +233,6 @@ transformers>=4.57.6 treescope>=0.1.10 typeguard>=2.13.3 typing-extensions>=4.15.0 -typing-inspect>=0.9.0 typing-inspection>=0.4.2 tzdata>=2025.3 ; sys_platform == 'emscripten' or sys_platform == 'win32' uritemplate>=4.2.0 From b741d4d68d52e25b475abce35eecc7381649f433 Mon Sep 17 00:00:00 2001 From: Dipannita Shaw Date: Sat, 7 Feb 2026 00:30:11 +0000 Subject: [PATCH 5/7] Revert GPU deps --- .../cuda12-requirements.txt | 190 +++++++++--------- 1 file changed, 94 insertions(+), 96 deletions(-) diff --git a/dependencies/requirements/generated_requirements/cuda12-requirements.txt b/dependencies/requirements/generated_requirements/cuda12-requirements.txt index 7ccaf4ec31..9879536ab1 100644 --- a/dependencies/requirements/generated_requirements/cuda12-requirements.txt +++ b/dependencies/requirements/generated_requirements/cuda12-requirements.txt @@ -4,26 +4,27 @@ absl-py>=2.3.1 aiofiles>=25.1.0 aiohappyeyeballs>=2.6.1 -aiohttp>=3.13.3 +aiohttp>=3.13.2 aiosignal>=1.4.0 annotated-doc>=0.0.4 annotated-types>=0.7.0 antlr4-python3-runtime>=4.9.3 -anyio>=4.12.1 +anyio>=4.11.0 aqtp>=0.9.0 array-record>=0.8.3 -astroid>=4.0.3 +astroid>=4.0.2 astunparse>=1.6.3 attrs>=25.4.0 auditwheel>=6.5.0 -black>=25.12.0 +black>=24.10.0 blobfile>=3.1.0 build>=1.3.0 +cachetools>=6.2.2 certifi>=2025.11.12 -cffi>=2.0.0 ; platform_python_implementation != 'PyPy' 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 @@ -31,83 +32,80 @@ cloudpickle>=3.1.2 clu>=0.0.12 colorama>=0.4.6 contourpy>=1.3.3 -coverage>=7.13.3 -cryptography>=46.0.4 +coverage>=7.12.0 cycler>=0.12.1 -datasets>=4.5.0 +datasets>=4.4.1 decorator>=5.2.1 dill>=0.4.0 distlib>=0.4.0 -distro>=1.9.0 dm-tree>=0.1.9 docstring-parser>=0.17.0 drjax>=0.1.4 editdistance>=0.8.1 -einops>=0.8.2 +einops>=0.8.1 einshape>=1.0 etils>=1.13.0 evaluate>=0.4.6 execnet>=2.1.2 -fastapi>=0.128.3 -filelock>=3.20.1 +fastapi>=0.122.0 +filelock>=3.20.0 flatbuffers>=25.9.23 -flax>=0.12.2 -fonttools>=4.61.1 +flax>=0.12.1 +fonttools>=4.60.1 frozenlist>=1.8.0 fsspec>=2025.10.0 gast>=0.6.0 gcsfs>=2025.10.0 -google-api-core>=2.29.0 -google-api-python-client>=2.189.0 -google-auth-httplib2>=0.3.0 -google-auth-oauthlib>=1.2.4 -google-auth>=2.48.0 -google-cloud-aiplatform>=1.136.0 -google-cloud-appengine-logging>=1.8.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.128.0 +google-cloud-appengine-logging>=1.7.0 google-cloud-audit-log>=0.4.0 -google-cloud-bigquery>=3.40.0 +google-cloud-bigquery>=3.38.0 google-cloud-core>=2.5.0 -google-cloud-logging>=3.13.0 -google-cloud-mldiagnostics>=0.5.10 -google-cloud-monitoring>=2.29.1 -google-cloud-resource-manager>=1.16.0 -google-cloud-storage>=3.9.0 -google-crc32c>=1.8.0 -google-genai>=1.62.0 -google-jetstream @ https://github.com/AI-Hypercomputer/JetStream/archive/29329e8e73820993f77cfc8efe34eb2a73f5de98.zip +google-cloud-logging>=3.12.1 +google-cloud-mldiagnostics>=0.5.5 +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.52.0 google-pasta>=0.2.0 google-resumable-media>=2.8.0 googleapis-common-protos>=1.72.0 grain>=0.2.15 grpc-google-iam-v1>=0.14.3 -grpcio-status>=1.76.0 +grpcio-status>=1.71.2 grpcio>=1.76.0 gviz-api>=1.10.0 h11>=0.16.0 h5py>=3.15.1 -hf-xet>=1.2.0 +hf-xet>=1.2.0 ; platform_machine == 'aarch64' or platform_machine == 'amd64' or platform_machine == 'arm64' or platform_machine == 'x86_64' httpcore>=1.0.9 -httplib2>=0.31.2 +httplib2>=0.31.0 httpx>=0.28.1 huggingface-hub>=0.36.0 -humanize>=4.15.0 +humanize>=4.14.0 hypothesis>=6.142.1 -identify>=2.6.16 +identify>=2.6.15 idna>=3.11 immutabledict>=4.2.2 importlab>=0.8.1 -importlib-metadata>=8.7.1 +importlib-metadata>=8.7.0 importlib-resources>=6.5.2 iniconfig>=2.3.0 isort>=7.0.0 -jaraco-functools>=4.4.0 -jax-cuda12-pjrt>=0.8.2 ; sys_platform == 'linux' -jax-cuda12-plugin>=0.8.2 ; sys_platform == 'linux' -jax>=0.8.3 -jaxlib>=0.8.2 -jaxtyping>=0.3.7 +jaraco-functools>=4.3.0 +jax-cuda12-pjrt>=0.8.1 ; sys_platform == 'linux' +jax-cuda12-plugin>=0.8.1 ; sys_platform == 'linux' +jax>=0.8.1 +jaxlib>=0.8.1 +jaxtyping>=0.3.3 jinja2>=3.1.6 -joblib>=1.5.3 +joblib>=1.5.2 jsonlines>=4.0.0 keras>=3.12.0 kiwisolver>=1.4.9 @@ -117,27 +115,26 @@ lxml>=6.0.2 markdown-it-py>=4.0.0 markdown>=3.10 markupsafe>=3.0.3 -matplotlib>=3.10.8 +matplotlib>=3.10.7 mccabe>=0.7.0 mdurl>=0.1.2 ml-collections>=1.1.0 ml-dtypes>=0.5.4 -ml-goodput-measurement>=0.0.16 -mlperf-logging @ https://github.com/mlcommons/logging/archive/38ab22670527888c8eb7825a4ece176fcc36a95d.zip +ml-goodput-measurement>=0.0.15 more-itertools>=10.8.0 mpmath>=1.3.0 msgpack>=1.1.2 msgspec>=0.20.0 -multidict>=6.7.1 +multidict>=6.7.0 multiprocess>=0.70.18 mypy-extensions>=1.1.0 namex>=0.1.0 nest-asyncio>=1.6.0 -networkx>=3.6.1 +networkx>=3.6 ninja>=1.13.0 nltk>=3.9.2 -nodeenv>=1.10.0 -numpy-typing-compat>=20251206.2.0 +nodeenv>=1.9.1 +numpy-typing-compat>=20250818.2.0 numpy>=2.0.2 nvidia-cublas-cu12>=12.9.1.4 ; sys_platform == 'linux' nvidia-cuda-cupti-cu12>=12.9.79 ; sys_platform == 'linux' @@ -153,33 +150,32 @@ 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.39.1 +opentelemetry-api>=1.38.0 opt-einsum>=3.4.0 -optax>=0.2.7 +optax>=0.2.6 optree>=0.18.0 -optype>=0.15.0 -orbax-checkpoint>=0.11.32 +optype>=0.14.0 +orbax-checkpoint>=0.11.28 packaging>=25.0 -pandas>=3.0.0 +pandas>=2.3.3 parameterized>=0.9.0 -pathspec>=1.0.4 -pathwaysutils>=0.1.4 +pathspec>=0.12.1 +pathwaysutils>=0.1.3 pillow>=12.0.0 -platformdirs>=4.5.1 +platformdirs>=4.5.0 pluggy>=1.6.0 portpicker>=1.6.0 -pre-commit>=4.5.1 -prometheus-client>=0.24.1 +pre-commit>=4.5.0 +prometheus-client>=0.23.1 promise>=2.3 propcache>=0.4.1 -proto-plus>=1.27.1 -protobuf>=6.33.5 +proto-plus>=1.26.1 +protobuf>=5.29.5 psutil>=7.1.3 -pyarrow>=23.0.0 +pyarrow>=22.0.0 pyasn1-modules>=0.4.2 -pyasn1>=0.6.2 +pyasn1>=0.6.1 pycnite>=2024.7.31 -pycparser>=3.0 ; implementation_name != 'PyPy' and platform_python_implementation != 'PyPy' pycryptodomex>=3.23.0 pydantic-core>=2.41.5 pydantic>=2.12.5 @@ -187,74 +183,76 @@ pydot>=4.0.1 pyelftools>=0.32 pyglove>=0.4.5 pygments>=2.19.2 -pyink>=25.12.0 -pylint>=4.0.4 +pyink>=24.10.1 +pylint>=4.0.3 pyparsing>=3.2.5 pyproject-hooks>=1.2.0 pytest-xdist>=3.8.0 pytest>=8.4.2 python-dateutil>=2.9.0.post0 -pytokens>=0.4.1 pytype>=2024.10.11 +pytz>=2025.2 pyyaml>=6.0.3 -qwix>=0.1.5 -regex>=2026.1.15 +qwix>=0.1.4 +regex>=2025.11.3 requests-oauthlib>=2.0.0 requests>=2.32.5 rich>=14.2.0 rsa>=4.9.1 safetensors>=0.7.0 -scipy-stubs>=1.16.3.3 +scipy-stubs>=1.16.3.0 scipy>=1.16.3 sentencepiece>=0.2.1 seqio>=0.0.20 setuptools>=80.9.0 +shapely>=2.1.2 shortuuid>=1.0.13 -simple-parsing>=0.1.8 +simple-parsing>=0.1.7 simplejson>=3.20.2 six>=1.17.0 sniffio>=1.3.1 sortedcontainers>=2.4.0 -starlette>=0.52.1 +starlette>=0.50.0 sympy>=1.14.0 tabulate>=0.9.0 -tenacity>=9.1.3 +tenacity>=9.1.2 tensorboard-data-server>=0.7.2 tensorboard-plugin-profile>=2.13.0 -tensorboard>=2.20.0 +tensorboard>=2.19.0 tensorboardx>=2.6.4 tensorflow-datasets>=4.9.9 -tensorflow-metadata>=1.17.3 -tensorflow-text>=2.20.0 -tensorflow>=2.20.0 -tensorstore>=0.1.80 +tensorflow-metadata>=1.17.2 +tensorflow-text>=2.19.0 +tensorflow>=2.19.1 +tensorstore>=0.1.79 termcolor>=3.2.0 tiktoken>=0.12.0 -tokamax>=0.0.10 -tokenizers>=0.22.2 +tokamax>=0.0.8 +tokenizers>=0.22.1 toml>=0.10.2 -tomlkit>=0.14.0 -tqdm>=4.67.3 -transformer-engine-cu12>=2.11.0 -transformer-engine-jax>=2.11.0 -transformer-engine>=2.11.0 -transformers>=4.57.6 +tomlkit>=0.13.3 +toolz>=1.1.0 +tqdm>=4.67.1 +transformer-engine-cu12>=2.9.0 +transformer-engine-jax>=2.9.0 +transformer-engine>=2.9.0 +transformers>=4.57.3 treescope>=0.1.10 typeguard>=2.13.3 typing-extensions>=4.15.0 typing-inspection>=0.4.2 -tzdata>=2025.3 ; sys_platform == 'emscripten' or sys_platform == 'win32' +tzdata>=2025.2 uritemplate>=4.2.0 -urllib3>=2.6.2 -uvicorn>=0.40.0 -virtualenv>=20.36.1 +urllib3>=2.5.0 +uvicorn>=0.38.0 +virtualenv>=20.35.4 wadler-lindig>=0.1.7 websockets>=15.0.1 -werkzeug>=3.1.4 -wheel>=0.46.1 -wrapt>=2.1.1 -xprof>=2.21.5 +werkzeug>=3.1.3 +wheel>=0.45.1 +wrapt>=2.0.1 +xprof>=2.21.1 xxhash>=3.6.0 yarl>=1.22.0 zipp>=3.23.0 -zstandard>=0.25.0 \ No newline at end of file +zstandard>=0.25.0 From 2d7570e07e7c276e191c08b776f9b244d658a51a Mon Sep 17 00:00:00 2001 From: Dipannita Shaw Date: Fri, 13 Feb 2026 23:00:56 +0000 Subject: [PATCH 6/7] Re-run seed-env with same commit & manually fix jax>0.8.3 --- .../tpu-requirements.txt | 32 +++++++++---------- 1 file changed, 16 insertions(+), 16 deletions(-) diff --git a/dependencies/requirements/generated_requirements/tpu-requirements.txt b/dependencies/requirements/generated_requirements/tpu-requirements.txt index 6cba264033..b3278e9cf5 100644 --- a/dependencies/requirements/generated_requirements/tpu-requirements.txt +++ b/dependencies/requirements/generated_requirements/tpu-requirements.txt @@ -12,12 +12,12 @@ antlr4-python3-runtime>=4.9.3 anyio>=4.12.1 aqtp>=0.9.0 array-record>=0.8.3 -astroid>=4.0.3 +astroid>=4.0.4 astunparse>=1.6.3 attrs>=25.4.0 auditwheel>=6.5.0 black>=25.12.0 -blobfile>=3.1.0 +blobfile>=3.2.0 build>=1.3.0 certifi>=2025.11.12 cffi>=2.0.0 ; platform_python_implementation != 'PyPy' @@ -31,8 +31,8 @@ cloudpickle>=3.1.2 clu>=0.0.12 colorama>=0.4.6 contourpy>=1.3.3 -coverage>=7.13.3 -cryptography>=46.0.4 +coverage>=7.13.4 +cryptography>=46.0.5 cycler>=0.12.1 datasets>=4.5.0 decorator>=5.2.1 @@ -48,7 +48,7 @@ einshape>=1.0 etils>=1.13.0 evaluate>=0.4.6 execnet>=2.1.2 -fastapi>=0.128.3 +fastapi>=0.129.0 filelock>=3.20.1 flatbuffers>=25.9.23 flax>=0.12.2 @@ -58,22 +58,22 @@ fsspec>=2025.10.0 gast>=0.6.0 gcsfs>=2025.10.0 google-api-core>=2.29.0 -google-api-python-client>=2.189.0 +google-api-python-client>=2.190.0 google-auth-httplib2>=0.3.0 google-auth-oauthlib>=1.2.4 google-auth>=2.48.0 -google-cloud-aiplatform>=1.136.0 +google-cloud-aiplatform>=1.137.0 google-cloud-appengine-logging>=1.8.0 google-cloud-audit-log>=0.4.0 -google-cloud-bigquery>=3.40.0 +google-cloud-bigquery>=3.40.1 google-cloud-core>=2.5.0 google-cloud-logging>=3.13.0 -google-cloud-mldiagnostics>=0.5.10 +google-cloud-mldiagnostics>=0.5.11 google-cloud-monitoring>=2.29.1 google-cloud-resource-manager>=1.16.0 google-cloud-storage>=3.9.0 google-crc32c>=1.8.0 -google-genai>=1.62.0 +google-genai>=1.63.0 google-jetstream @ https://github.com/AI-Hypercomputer/JetStream/archive/29329e8e73820993f77cfc8efe34eb2a73f5de98.zip google-pasta>=0.2.0 google-resumable-media>=2.8.0 @@ -96,20 +96,20 @@ humanize>=4.15.0 hypothesis>=6.142.1 identify>=2.6.16 idna>=3.11 -immutabledict>=4.2.2 +immutabledict>=4.3.0 importlab>=0.8.1 importlib-metadata>=8.7.1 importlib-resources>=6.5.2 iniconfig>=2.3.0 isort>=7.0.0 jaraco-functools>=4.4.0 -jax>=0.8.3 +jax>=0.8.2 jaxlib>=0.8.2 jaxtyping>=0.3.7 jinja2>=3.1.6 joblib>=1.5.3 jsonlines>=4.0.0 -kagglehub>=0.4.3 +kagglehub>=1.0.0 kagglesdk>=0.1.15 keras>=3.12.0 kiwisolver>=1.4.9 @@ -158,7 +158,7 @@ parameterized>=0.9.0 pathspec>=1.0.4 pathwaysutils>=0.1.4 pillow>=12.0.0 -platformdirs>=4.5.1 +platformdirs>=4.7.1 pluggy>=1.6.0 portpicker>=1.6.0 pre-commit>=4.5.1 @@ -212,7 +212,7 @@ sortedcontainers>=2.4.0 starlette>=0.52.1 sympy>=1.14.0 tabulate>=0.9.0 -tenacity>=9.1.3 +tenacity>=9.1.4 tensorboard-data-server>=0.7.2 tensorboard-plugin-profile>=2.13.0 tensorboard>=2.20.0 @@ -244,7 +244,7 @@ websockets>=15.0.1 werkzeug>=3.1.4 wheel>=0.46.1 wrapt>=2.1.1 -xprof>=2.21.5 +xprof>=2.21.6 xxhash>=3.6.0 yarl>=1.22.0 zipp>=3.23.0 From 17a2e076c7f7515c753fc0bfe419418e4ddad509 Mon Sep 17 00:00:00 2001 From: Dipannita Shaw Date: Sat, 14 Feb 2026 02:43:31 +0000 Subject: [PATCH 7/7] . --- .../base_requirements/requirements.txt | 2 +- .../cuda12-requirements.txt | 190 +++++++++--------- .../tpu-requirements.txt | 32 +-- 3 files changed, 113 insertions(+), 111 deletions(-) diff --git a/dependencies/requirements/base_requirements/requirements.txt b/dependencies/requirements/base_requirements/requirements.txt index 27941b0ca9..8add067a7f 100644 --- a/dependencies/requirements/base_requirements/requirements.txt +++ b/dependencies/requirements/base_requirements/requirements.txt @@ -44,4 +44,4 @@ tokamax!=0.1.0 transformers qwix google-jetstream @ https://github.com/AI-Hypercomputer/JetStream/archive/29329e8e73820993f77cfc8efe34eb2a73f5de98.zip -mlperf-logging @ https://github.com/mlcommons/logging/archive/38ab22670527888c8eb7825a4ece176fcc36a95d.zip +mlperf-logging @ https://github.com/mlcommons/logging/archive/38ab22670527888c8eb7825a4ece176fcc36a95d.zip \ No newline at end of file diff --git a/dependencies/requirements/generated_requirements/cuda12-requirements.txt b/dependencies/requirements/generated_requirements/cuda12-requirements.txt index 9879536ab1..7ccaf4ec31 100644 --- a/dependencies/requirements/generated_requirements/cuda12-requirements.txt +++ b/dependencies/requirements/generated_requirements/cuda12-requirements.txt @@ -4,27 +4,26 @@ absl-py>=2.3.1 aiofiles>=25.1.0 aiohappyeyeballs>=2.6.1 -aiohttp>=3.13.2 +aiohttp>=3.13.3 aiosignal>=1.4.0 annotated-doc>=0.0.4 annotated-types>=0.7.0 antlr4-python3-runtime>=4.9.3 -anyio>=4.11.0 +anyio>=4.12.1 aqtp>=0.9.0 array-record>=0.8.3 -astroid>=4.0.2 +astroid>=4.0.3 astunparse>=1.6.3 attrs>=25.4.0 auditwheel>=6.5.0 -black>=24.10.0 +black>=25.12.0 blobfile>=3.1.0 build>=1.3.0 -cachetools>=6.2.2 certifi>=2025.11.12 +cffi>=2.0.0 ; platform_python_implementation != 'PyPy' 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 @@ -32,80 +31,83 @@ cloudpickle>=3.1.2 clu>=0.0.12 colorama>=0.4.6 contourpy>=1.3.3 -coverage>=7.12.0 +coverage>=7.13.3 +cryptography>=46.0.4 cycler>=0.12.1 -datasets>=4.4.1 +datasets>=4.5.0 decorator>=5.2.1 dill>=0.4.0 distlib>=0.4.0 +distro>=1.9.0 dm-tree>=0.1.9 docstring-parser>=0.17.0 drjax>=0.1.4 editdistance>=0.8.1 -einops>=0.8.1 +einops>=0.8.2 einshape>=1.0 etils>=1.13.0 evaluate>=0.4.6 execnet>=2.1.2 -fastapi>=0.122.0 -filelock>=3.20.0 +fastapi>=0.128.3 +filelock>=3.20.1 flatbuffers>=25.9.23 -flax>=0.12.1 -fonttools>=4.60.1 +flax>=0.12.2 +fonttools>=4.61.1 frozenlist>=1.8.0 fsspec>=2025.10.0 gast>=0.6.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.128.0 -google-cloud-appengine-logging>=1.7.0 +google-api-core>=2.29.0 +google-api-python-client>=2.189.0 +google-auth-httplib2>=0.3.0 +google-auth-oauthlib>=1.2.4 +google-auth>=2.48.0 +google-cloud-aiplatform>=1.136.0 +google-cloud-appengine-logging>=1.8.0 google-cloud-audit-log>=0.4.0 -google-cloud-bigquery>=3.38.0 +google-cloud-bigquery>=3.40.0 google-cloud-core>=2.5.0 -google-cloud-logging>=3.12.1 -google-cloud-mldiagnostics>=0.5.5 -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.52.0 +google-cloud-logging>=3.13.0 +google-cloud-mldiagnostics>=0.5.10 +google-cloud-monitoring>=2.29.1 +google-cloud-resource-manager>=1.16.0 +google-cloud-storage>=3.9.0 +google-crc32c>=1.8.0 +google-genai>=1.62.0 +google-jetstream @ https://github.com/AI-Hypercomputer/JetStream/archive/29329e8e73820993f77cfc8efe34eb2a73f5de98.zip google-pasta>=0.2.0 google-resumable-media>=2.8.0 googleapis-common-protos>=1.72.0 grain>=0.2.15 grpc-google-iam-v1>=0.14.3 -grpcio-status>=1.71.2 +grpcio-status>=1.76.0 grpcio>=1.76.0 gviz-api>=1.10.0 h11>=0.16.0 h5py>=3.15.1 -hf-xet>=1.2.0 ; platform_machine == 'aarch64' or platform_machine == 'amd64' or platform_machine == 'arm64' or platform_machine == 'x86_64' +hf-xet>=1.2.0 httpcore>=1.0.9 -httplib2>=0.31.0 +httplib2>=0.31.2 httpx>=0.28.1 huggingface-hub>=0.36.0 -humanize>=4.14.0 +humanize>=4.15.0 hypothesis>=6.142.1 -identify>=2.6.15 +identify>=2.6.16 idna>=3.11 immutabledict>=4.2.2 importlab>=0.8.1 -importlib-metadata>=8.7.0 +importlib-metadata>=8.7.1 importlib-resources>=6.5.2 iniconfig>=2.3.0 isort>=7.0.0 -jaraco-functools>=4.3.0 -jax-cuda12-pjrt>=0.8.1 ; sys_platform == 'linux' -jax-cuda12-plugin>=0.8.1 ; sys_platform == 'linux' -jax>=0.8.1 -jaxlib>=0.8.1 -jaxtyping>=0.3.3 +jaraco-functools>=4.4.0 +jax-cuda12-pjrt>=0.8.2 ; sys_platform == 'linux' +jax-cuda12-plugin>=0.8.2 ; sys_platform == 'linux' +jax>=0.8.3 +jaxlib>=0.8.2 +jaxtyping>=0.3.7 jinja2>=3.1.6 -joblib>=1.5.2 +joblib>=1.5.3 jsonlines>=4.0.0 keras>=3.12.0 kiwisolver>=1.4.9 @@ -115,26 +117,27 @@ lxml>=6.0.2 markdown-it-py>=4.0.0 markdown>=3.10 markupsafe>=3.0.3 -matplotlib>=3.10.7 +matplotlib>=3.10.8 mccabe>=0.7.0 mdurl>=0.1.2 ml-collections>=1.1.0 ml-dtypes>=0.5.4 -ml-goodput-measurement>=0.0.15 +ml-goodput-measurement>=0.0.16 +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.20.0 -multidict>=6.7.0 +multidict>=6.7.1 multiprocess>=0.70.18 mypy-extensions>=1.1.0 namex>=0.1.0 nest-asyncio>=1.6.0 -networkx>=3.6 +networkx>=3.6.1 ninja>=1.13.0 nltk>=3.9.2 -nodeenv>=1.9.1 -numpy-typing-compat>=20250818.2.0 +nodeenv>=1.10.0 +numpy-typing-compat>=20251206.2.0 numpy>=2.0.2 nvidia-cublas-cu12>=12.9.1.4 ; sys_platform == 'linux' nvidia-cuda-cupti-cu12>=12.9.79 ; sys_platform == 'linux' @@ -150,32 +153,33 @@ 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 +opentelemetry-api>=1.39.1 opt-einsum>=3.4.0 -optax>=0.2.6 +optax>=0.2.7 optree>=0.18.0 -optype>=0.14.0 -orbax-checkpoint>=0.11.28 +optype>=0.15.0 +orbax-checkpoint>=0.11.32 packaging>=25.0 -pandas>=2.3.3 +pandas>=3.0.0 parameterized>=0.9.0 -pathspec>=0.12.1 -pathwaysutils>=0.1.3 +pathspec>=1.0.4 +pathwaysutils>=0.1.4 pillow>=12.0.0 -platformdirs>=4.5.0 +platformdirs>=4.5.1 pluggy>=1.6.0 portpicker>=1.6.0 -pre-commit>=4.5.0 -prometheus-client>=0.23.1 +pre-commit>=4.5.1 +prometheus-client>=0.24.1 promise>=2.3 propcache>=0.4.1 -proto-plus>=1.26.1 -protobuf>=5.29.5 +proto-plus>=1.27.1 +protobuf>=6.33.5 psutil>=7.1.3 -pyarrow>=22.0.0 +pyarrow>=23.0.0 pyasn1-modules>=0.4.2 -pyasn1>=0.6.1 +pyasn1>=0.6.2 pycnite>=2024.7.31 +pycparser>=3.0 ; implementation_name != 'PyPy' and platform_python_implementation != 'PyPy' pycryptodomex>=3.23.0 pydantic-core>=2.41.5 pydantic>=2.12.5 @@ -183,76 +187,74 @@ pydot>=4.0.1 pyelftools>=0.32 pyglove>=0.4.5 pygments>=2.19.2 -pyink>=24.10.1 -pylint>=4.0.3 +pyink>=25.12.0 +pylint>=4.0.4 pyparsing>=3.2.5 pyproject-hooks>=1.2.0 pytest-xdist>=3.8.0 pytest>=8.4.2 python-dateutil>=2.9.0.post0 +pytokens>=0.4.1 pytype>=2024.10.11 -pytz>=2025.2 pyyaml>=6.0.3 -qwix>=0.1.4 -regex>=2025.11.3 +qwix>=0.1.5 +regex>=2026.1.15 requests-oauthlib>=2.0.0 requests>=2.32.5 rich>=14.2.0 rsa>=4.9.1 safetensors>=0.7.0 -scipy-stubs>=1.16.3.0 +scipy-stubs>=1.16.3.3 scipy>=1.16.3 sentencepiece>=0.2.1 seqio>=0.0.20 setuptools>=80.9.0 -shapely>=2.1.2 shortuuid>=1.0.13 -simple-parsing>=0.1.7 +simple-parsing>=0.1.8 simplejson>=3.20.2 six>=1.17.0 sniffio>=1.3.1 sortedcontainers>=2.4.0 -starlette>=0.50.0 +starlette>=0.52.1 sympy>=1.14.0 tabulate>=0.9.0 -tenacity>=9.1.2 +tenacity>=9.1.3 tensorboard-data-server>=0.7.2 tensorboard-plugin-profile>=2.13.0 -tensorboard>=2.19.0 +tensorboard>=2.20.0 tensorboardx>=2.6.4 tensorflow-datasets>=4.9.9 -tensorflow-metadata>=1.17.2 -tensorflow-text>=2.19.0 -tensorflow>=2.19.1 -tensorstore>=0.1.79 +tensorflow-metadata>=1.17.3 +tensorflow-text>=2.20.0 +tensorflow>=2.20.0 +tensorstore>=0.1.80 termcolor>=3.2.0 tiktoken>=0.12.0 -tokamax>=0.0.8 -tokenizers>=0.22.1 +tokamax>=0.0.10 +tokenizers>=0.22.2 toml>=0.10.2 -tomlkit>=0.13.3 -toolz>=1.1.0 -tqdm>=4.67.1 -transformer-engine-cu12>=2.9.0 -transformer-engine-jax>=2.9.0 -transformer-engine>=2.9.0 -transformers>=4.57.3 +tomlkit>=0.14.0 +tqdm>=4.67.3 +transformer-engine-cu12>=2.11.0 +transformer-engine-jax>=2.11.0 +transformer-engine>=2.11.0 +transformers>=4.57.6 treescope>=0.1.10 typeguard>=2.13.3 typing-extensions>=4.15.0 typing-inspection>=0.4.2 -tzdata>=2025.2 +tzdata>=2025.3 ; sys_platform == 'emscripten' or sys_platform == 'win32' uritemplate>=4.2.0 -urllib3>=2.5.0 -uvicorn>=0.38.0 -virtualenv>=20.35.4 +urllib3>=2.6.2 +uvicorn>=0.40.0 +virtualenv>=20.36.1 wadler-lindig>=0.1.7 websockets>=15.0.1 -werkzeug>=3.1.3 -wheel>=0.45.1 -wrapt>=2.0.1 -xprof>=2.21.1 +werkzeug>=3.1.4 +wheel>=0.46.1 +wrapt>=2.1.1 +xprof>=2.21.5 xxhash>=3.6.0 yarl>=1.22.0 zipp>=3.23.0 -zstandard>=0.25.0 +zstandard>=0.25.0 \ No newline at end of file diff --git a/dependencies/requirements/generated_requirements/tpu-requirements.txt b/dependencies/requirements/generated_requirements/tpu-requirements.txt index b3278e9cf5..6cba264033 100644 --- a/dependencies/requirements/generated_requirements/tpu-requirements.txt +++ b/dependencies/requirements/generated_requirements/tpu-requirements.txt @@ -12,12 +12,12 @@ antlr4-python3-runtime>=4.9.3 anyio>=4.12.1 aqtp>=0.9.0 array-record>=0.8.3 -astroid>=4.0.4 +astroid>=4.0.3 astunparse>=1.6.3 attrs>=25.4.0 auditwheel>=6.5.0 black>=25.12.0 -blobfile>=3.2.0 +blobfile>=3.1.0 build>=1.3.0 certifi>=2025.11.12 cffi>=2.0.0 ; platform_python_implementation != 'PyPy' @@ -31,8 +31,8 @@ cloudpickle>=3.1.2 clu>=0.0.12 colorama>=0.4.6 contourpy>=1.3.3 -coverage>=7.13.4 -cryptography>=46.0.5 +coverage>=7.13.3 +cryptography>=46.0.4 cycler>=0.12.1 datasets>=4.5.0 decorator>=5.2.1 @@ -48,7 +48,7 @@ einshape>=1.0 etils>=1.13.0 evaluate>=0.4.6 execnet>=2.1.2 -fastapi>=0.129.0 +fastapi>=0.128.3 filelock>=3.20.1 flatbuffers>=25.9.23 flax>=0.12.2 @@ -58,22 +58,22 @@ fsspec>=2025.10.0 gast>=0.6.0 gcsfs>=2025.10.0 google-api-core>=2.29.0 -google-api-python-client>=2.190.0 +google-api-python-client>=2.189.0 google-auth-httplib2>=0.3.0 google-auth-oauthlib>=1.2.4 google-auth>=2.48.0 -google-cloud-aiplatform>=1.137.0 +google-cloud-aiplatform>=1.136.0 google-cloud-appengine-logging>=1.8.0 google-cloud-audit-log>=0.4.0 -google-cloud-bigquery>=3.40.1 +google-cloud-bigquery>=3.40.0 google-cloud-core>=2.5.0 google-cloud-logging>=3.13.0 -google-cloud-mldiagnostics>=0.5.11 +google-cloud-mldiagnostics>=0.5.10 google-cloud-monitoring>=2.29.1 google-cloud-resource-manager>=1.16.0 google-cloud-storage>=3.9.0 google-crc32c>=1.8.0 -google-genai>=1.63.0 +google-genai>=1.62.0 google-jetstream @ https://github.com/AI-Hypercomputer/JetStream/archive/29329e8e73820993f77cfc8efe34eb2a73f5de98.zip google-pasta>=0.2.0 google-resumable-media>=2.8.0 @@ -96,20 +96,20 @@ humanize>=4.15.0 hypothesis>=6.142.1 identify>=2.6.16 idna>=3.11 -immutabledict>=4.3.0 +immutabledict>=4.2.2 importlab>=0.8.1 importlib-metadata>=8.7.1 importlib-resources>=6.5.2 iniconfig>=2.3.0 isort>=7.0.0 jaraco-functools>=4.4.0 -jax>=0.8.2 +jax>=0.8.3 jaxlib>=0.8.2 jaxtyping>=0.3.7 jinja2>=3.1.6 joblib>=1.5.3 jsonlines>=4.0.0 -kagglehub>=1.0.0 +kagglehub>=0.4.3 kagglesdk>=0.1.15 keras>=3.12.0 kiwisolver>=1.4.9 @@ -158,7 +158,7 @@ parameterized>=0.9.0 pathspec>=1.0.4 pathwaysutils>=0.1.4 pillow>=12.0.0 -platformdirs>=4.7.1 +platformdirs>=4.5.1 pluggy>=1.6.0 portpicker>=1.6.0 pre-commit>=4.5.1 @@ -212,7 +212,7 @@ sortedcontainers>=2.4.0 starlette>=0.52.1 sympy>=1.14.0 tabulate>=0.9.0 -tenacity>=9.1.4 +tenacity>=9.1.3 tensorboard-data-server>=0.7.2 tensorboard-plugin-profile>=2.13.0 tensorboard>=2.20.0 @@ -244,7 +244,7 @@ websockets>=15.0.1 werkzeug>=3.1.4 wheel>=0.46.1 wrapt>=2.1.1 -xprof>=2.21.6 +xprof>=2.21.5 xxhash>=3.6.0 yarl>=1.22.0 zipp>=3.23.0