From d81e2cf7e193a2d4e0a29bb4cb279e0746a7e152 Mon Sep 17 00:00:00 2001
From: Angel Mau <angelmau@google.com>
Date: Tue, 14 Apr 2026 08:48:32 -0700
Subject: [PATCH] Add orbax-checkpoint to tunix dependencies.

PiperOrigin-RevId: 899610145
---
 pyproject.toml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/pyproject.toml b/pyproject.toml
index f01e0a914..3844bf302 100644
--- 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