Add orbax-checkpoint to tunix dependencies.

angel-core · The tunix Authors · commit d81e2cf7e193 · 2026-05-01T14:26:13.000-07:00
PiperOrigin-RevId: 899610145
diff --git a/pyproject.toml b/pyproject.toml
@@ -31,6 +31,7 @@ dependencies = [
   "kagglehub",
   "numba",
   "omegaconf", # CLI config
+  "orbax-checkpoint>=0.11.37",
   "pillow",  # Image processing
   "pylatexenc",  # Eval result parsing
   "python-dotenv",  # Huggingface API key
