Update generated requirements for JAX 0.9.0

Author: SurbhiJainUSC

src/dependencies/requirements/base_requirements/requirements.txt

@@ -1,8 +1,9 @@
 absl-py
 aqtp
 array-record
+chex
 cloud-accelerator-diagnostics
-cloud-tpu-diagnostics
+cloud-tpu-diagnostics!=1.1.14
 datasets
 drjax
 flax
@@ -40,7 +41,7 @@ tensorflow-datasets
 tensorflow-text
 tensorflow
 tiktoken
-tokamax
+tokamax!=0.1.0
 transformers
 uvloop
 qwix

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

@@ -4,23 +4,23 @@
 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.4
 astunparse>=1.6.3
 attrs>=25.4.0
-auditwheel>=6.5.0
-black>=24.10.0
-blobfile>=3.1.0
-build>=1.3.0
-cachetools>=6.2.2
-certifi>=2025.11.12
+auditwheel>=6.6.0
+black>=25.12.0
+blobfile>=3.2.0
+build>=1.4.0
+certifi>=2026.1.4
+cffi>=2.0.0 ; platform_python_implementation != 'PyPy'
 cfgv>=3.5.0
 charset-normalizer>=3.4.4
 cheroot>=11.1.2
@@ -32,221 +32,231 @@ cloudpickle>=3.1.2
 clu>=0.0.12
 colorama>=0.4.6
 contourpy>=1.3.3
-coverage>=7.12.0
+coverage>=7.13.4
+cryptography>=46.0.5
 cycler>=0.12.1
-dacite>=1.9.2
-datasets>=4.4.1
+dataclasses-json>=0.6.7
+datasets>=4.7.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
-flatbuffers>=25.9.23
-flax>=0.12.1
-fonttools>=4.60.1
+fastapi>=0.135.1
+filelock>=3.20.3
+flatbuffers>=25.12.19
+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
+fsspec>=2026.1.0
+gast>=0.7.0
+gcsfs>=2026.1.0
+google-api-core>=2.30.0
+google-api-python-client>=2.192.0
+google-auth-httplib2>=0.3.0
+google-auth-oauthlib>=1.3.0
+google-auth>=2.49.1
+google-cloud-aiplatform>=1.141.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.1
 google-cloud-core>=2.5.0
-google-cloud-logging>=3.12.1
-google-cloud-mldiagnostics>=0.5.10
-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.14.0
+google-cloud-mldiagnostics>=1.0.0
+google-cloud-monitoring>=2.29.1
+google-cloud-resource-manager>=1.16.0
+google-cloud-storage-control>=1.10.0
+google-cloud-storage>=3.9.0
+google-crc32c>=1.8.0
+google-genai>=1.67.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
+googleapis-common-protos>=1.73.0
+grain>=0.2.16
 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.4.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.6.0
+humanize>=4.15.0
 hypothesis>=6.142.1
-identify>=2.6.15
+identify>=2.6.17
 idna>=3.11
-immutabledict>=4.2.2
+immutabledict>=4.3.1
 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
+isort>=8.0.1
+jaraco-functools>=4.4.0
+jax>=0.9.0
+jaxlib>=0.9.0
+jaxtyping>=0.3.9
 jinja2>=3.1.6
-joblib>=1.5.2
+joblib>=1.5.3
 jsonlines>=4.0.0
-kagglehub>=0.3.13
-keras>=3.12.0
+kagglehub>=1.0.0
+kagglesdk>=0.1.16
+keras>=3.13.1
 kiwisolver>=1.4.9
+latex2sympy2-extended>=1.11.0
 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.34 ; 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
+markdown>=3.10.1
 markupsafe>=3.0.3
+marshmallow>=3.26.2
 math-verify>=0.9.0
-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
-numba>=0.62.1
-numpy-typing-compat>=20250818.2.0
+nltk>=3.9.3
+nodeenv>=1.10.0
+numba>=0.64.0
+numpy-typing-compat>=20251206.2.0
 numpy>=2.0.2
+nvidia-cuda-cccl>=13.1.115
 oauthlib>=3.3.1
 omegaconf>=2.3.0
-opentelemetry-api>=1.38.0
+opentelemetry-api>=1.40.0
 opt-einsum>=3.4.0
-optax>=0.2.6
+optax>=0.2.7
 optree>=0.18.0
-optype>=0.14.0
+optype>=0.15.0
 orbax-checkpoint>=0.11.33
-packaging>=25.0
-pandas>=2.3.3
+orbax-export>=0.0.8
+packaging>=26.0
+pandas>=3.0.1
 parameterized>=0.9.0
-pathspec>=0.12.1
-pathwaysutils>=0.1.4
-pillow>=12.0.0
-platformdirs>=4.5.0
+pathspec>=1.0.4
+pathwaysutils>=0.1.6
+pillow>=12.1.0
+platformdirs>=4.9.4
 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
-psutil>=7.1.3
-pyarrow>=22.0.0
+proto-plus>=1.27.1
+protobuf>=6.33.5
+psutil>=7.2.1
+pyarrow>=23.0.1
 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
 pydot>=4.0.1
 pyelftools>=0.32
 pyglove>=0.4.5
 pygments>=2.19.2
-pyink>=24.10.1
-pylint>=4.0.3
-pyparsing>=3.2.5
+pyink>=25.12.0
+pylint>=4.0.5
+pyparsing>=3.3.2
 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
+python-dotenv>=1.2.2
+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.2.28
 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>=1.16.3
+scipy-stubs>=1.17.0.1
+scipy>=1.17.0
 sentencepiece>=0.2.1
 seqio>=0.0.20
-setuptools>=80.9.0
-shapely>=2.1.2
+setuptools>=80.10.1
+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
 sortedcontainers>=2.4.0
-starlette>=0.50.0
+starlette>=0.52.1
 sympy>=1.14.0
-tabulate>=0.9.0
-tenacity>=9.1.2
+tabulate>=0.10.0
+tenacity>=9.1.4
 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
-termcolor>=3.2.0
+tensorflow-metadata>=1.17.3
+tensorflow-text>=2.20.1
+tensorflow>=2.20.0
+tensorstore>=0.1.80
+termcolor>=3.3.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
+tomlkit>=0.14.0
 toolz>=1.1.0
-tqdm>=4.67.1
-transformers>=4.57.3
+tqdm>=4.67.3
+transformers>=5.3.0
 treescope>=0.1.10
 typeguard>=2.13.3
+typer>=0.24.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
-uvloop>=0.19.0
-virtualenv>=20.35.4
+urllib3>=2.6.3
+uvicorn>=0.41.0
+uvloop>=0.22.1
+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
+websockets>=16.0
+werkzeug>=3.1.5
+wheel>=0.46.2
+wrapt>=2.1.2
+xprof>=2.22.0
 xxhash>=3.6.0
-yarl>=1.22.0
+yarl>=1.23.0
 zipp>=3.23.0
-zstandard>=0.25.0
+zstandard>=0.25.0