Update Orbax dependency for MaxText CUDA 12.

This change updates orbax-checkpoint to version 0.12.1 and adds it to the extra dependencies. It also includes version bumps for several other packages in the generated requirements, after running respective command for generated dep.

PiperOrigin-RevId: 949735177

Author: angel-core

src/dependencies/extra_deps/cuda12_overrides.txt

@@ -1,3 +1,4 @@
 datasets>=4.8.5
 fsspec==2026.2.0
 gcsfs==2026.2.0
+orbax-checkpoint>=0.12.1

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

@@ -30,7 +30,7 @@ cloudpickle>=3.1.2
 clu>=0.0.12
 colorama>=0.4.6
 contourpy>=1.3.3
-coverage>=7.14.3
+coverage>=7.15.0
 cryptography>=49.0.0
 cycler>=0.12.1
 datasets>=5.0.0
@@ -84,11 +84,11 @@ grpcio-status>=1.80.0
 gviz-api>=1.10.0
 h11>=0.16.0
 h5py>=3.14.0
-hf-xet>=1.5.1 ; platform_machine == 'AMD64' or platform_machine == 'aarch64' or platform_machine == 'amd64' or platform_machine == 'arm64' or platform_machine == 'x86_64'
+hf-xet>=1.5.1 ; platform_machine == 'AMD64' or platform_machine == 'aarch64' or platform_machine == 'x86_64'
 httpcore>=1.0.9
 httplib2>=0.32.0
 httpx>=0.28.1
-huggingface-hub>=1.21.0
+huggingface-hub>=1.22.0
 humanize>=4.16.0
 hypothesis>=6.142.1
 identify>=2.6.19
@@ -158,7 +158,7 @@ opt-einsum>=3.4.0
 optax>=0.2.8
 optree>=0.19.0
 optype>=0.17.0
-orbax-checkpoint>=0.11.39
+orbax-checkpoint>=0.12.1
 packaging>=26.1
 pandas>=3.0.3
 parameterized>=0.9.0
@@ -169,6 +169,7 @@ platformdirs>=4.10.0
 pluggy>=1.6.0
 portpicker>=1.6.0
 pre-commit>=4.6.0
+prometheus-client>=0.20.0
 promise>=2.3
 propcache>=0.5.2
 proto-plus>=1.28.0
@@ -195,7 +196,7 @@ pyproject-hooks>=1.2.0
 pytest>=8.4.2
 pytest-xdist>=3.8.0
 python-dateutil>=2.9.0.post0
-python-discovery>=1.4.2
+python-discovery>=1.4.3
 pytokens>=0.4.1
 pytype>=2024.10.11
 pyyaml>=6.0.3
@@ -229,7 +230,7 @@ tensorflow>=2.20.0
 tensorflow-datasets>=4.9.10
 tensorflow-metadata>=1.21.0
 tensorflow-text>=2.20.1
-tensorstore>=0.1.82
+tensorstore>=0.1.84
 termcolor>=3.3.0
 tiktoken>=0.13.0
 tokamax>=0.0.12
@@ -241,24 +242,24 @@ tqdm>=4.68.3
 transformer-engine>=2.16.1
 transformer-engine-cu12>=2.16.1
 transformer-engine-jax>=2.16.1
-transformers>=5.12.1
+transformers>=5.13.0
 treescope>=0.1.10
 typeguard>=2.13.3
-typer>=0.25.1
+typer>=0.26.8
 typing-extensions>=4.15.0
 typing-inspection>=0.4.2
 tzdata>=2026.2 ; sys_platform == 'emscripten' or sys_platform == 'win32'
 uritemplate>=4.2.0
 urllib3>=2.6.3
-uvicorn>=0.49.0
+uvicorn>=0.50.2
 uvloop>=0.22.1
-virtualenv>=21.5.1
+virtualenv>=21.5.2
 wadler-lindig>=0.1.7
 websockets>=16.0
 werkzeug>=3.1.8
 wheel>=0.46.3
 wrapt>=2.2.2
-xxhash>=3.8.0
+xxhash>=3.8.1
 yarl>=1.24.2
 zipp>=3.23.1
 zstandard>=0.25.0