diff --git a/pyproject.toml b/pyproject.toml
index 358d26a..1a6a8f6 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -19,7 +19,7 @@ dependencies = [
     "flax==0.10.2",
     "ml_dtypes==0.4.0",
     "optax==0.2.4",
-    "orbax-checkpoint==0.9.1",
+    "orbax-checkpoint==0.10.1",
     "orbax-export==0.0.6",
 ]