Bump orbax-checkpoint from 0.9.1 to 0.10.1

Bumps [orbax-checkpoint](https://github.com/google/orbax) from 0.9.1 to 0.10.1.
- [Release notes](https://github.com/google/orbax/releases)
- [Changelog](https://github.com/google/orbax/blob/main/CHANGELOG.md)
- [Commits](google/orbax@v0.9.1...v0.10.1)

---
updated-dependencies:
- dependency-name: orbax-checkpoint
  dependency-type: direct:production
  update-type: version-update:semver-minor
...

Signed-off-by: dependabot[bot] <[email protected]>

pyproject.toml

@@ -19,7 +19,7 @@ dependencies = [
     "flax==0.10.1",
     "ml_dtypes==0.4.0",
     "optax==0.2.4",
-    "orbax-checkpoint==0.9.1",
+    "orbax-checkpoint==0.10.1",
     "orbax-export==0.0.6",
 ]