Fix 413751357 updating orbax version which is lowering the JAX version

parambole · parambole · commit fccb25a33539 · 2025-04-25T23:13:48.000Z
diff --git a/requirements.txt b/requirements.txt
@@ -24,7 +24,7 @@ tensorflow-datasets>=4.9.6
 ruff>=0.1.5,<=0.2
 git+https://github.com/mlperf/logging.git
 opencv-python-headless==4.10.0.84
-orbax-checkpoint==0.10.3
+orbax-checkpoint>=0.10.3
 tokenizers==0.21.0
 huggingface_hub==0.30.2
 transformers==4.48.1
diff --git a/requirements_with_jax_stable_stack.txt b/requirements_with_jax_stable_stack.txt
@@ -14,7 +14,7 @@ jaxlib>=0.4.30
 Jinja2
 opencv-python-headless==4.10.0.84
 optax>=0.2.3
-orbax-checkpoint==0.10.3
+orbax-checkpoint>=0.10.3
 parameterized
 Pillow
 pyink
