orbax-checkpoint
Orbax Checkpoint
Why this rank:Strong adoptionRecent releaseHealthy release cadence
Description
# Orbax Checkpointing `pip install orbax-checkpoint` (latest PyPi release) OR `pip install 'git+https://github.com/google/orbax/#subdirectory=checkpoint'` (from this repository, at HEAD) `import orbax.checkpoint` Orbax includes a checkpointing library oriented towards JAX users, supporting a variety of different features required by different frameworks, including asynchronous checkpointing, various types, and various storage formats. We aim to provide a highly customizable and composable API which maximizes flexibility for diverse use cases.
Release History
| Version | Changes | Urgency | Date |
|---|---|---|---|
| v0.12.0 | ### Added - Add `LatestDuration` preservation policy to support duration-based (TTL) checkpoint retention. - Add `EveryNStepsClosest` preservation policy, which preserves checkpoints at steps closest to multiples of `interval_steps`. - Add `SerializationStatusCallback` for save operations to track transfer and disk write events. While primarily intended as lightweight progress handles, this callback system allows the potential to inject custom logic for advanced monitoring and custom actions | High | 6/2/2026 |
| v0.11.40 | ### Removed - #v1 Remove numpy and pytorch layouts, as these are insecure and not worth the effort in maintaining. Safetensors is instead the recommended conversion case. - #v1 Remove `LeafHandler` as a user-exposed layer (it remains as an internal layer). ### Added - #v1 Allow a context to be default-configured for all `Checkpointer` operations. ### Changed - Upgrade to TensorStore>=0.1.84 (and Python>=3.11 as a consequence.) | High | 5/19/2026 |
| v0.11.39 | ### Added - Tensorstore non-atomic file I/O locking on local/POSIX-compatible filesystems to avoid unnecessary renames. | High | 5/6/2026 |
| v0.11.37 | ### Added - Add colocated runtime helpers for Pathways MTC. - #v1 Centralize `StorageOptions` into `ArrayOptions` and implement field-level merging. - Add Patch for Pathways CPU ids. - Add sidecar MTC backend support - Add Colocated Python Sidecar MTC Checkpoint Manager - Add Colocated Python Orchestration handling - #v1 Add `LeafHandler` as a `CheckpointableHandler`, so that ordinary PyTree leaves can also be saved as individual checkpointables. - Move MTC files to multi_tier_checkp | High | 5/5/2026 |
| 0.11.36 | Imported from PyPI (0.11.36) | Low | 4/21/2026 |
| v0.11.36 | ### Added - #v1 Add auto loading functionality to `load_pytree` and `pytree_metadata`. | Medium | 4/14/2026 |
| v0.11.36 | ### Added - #v1 Add auto loading functionality to `load_pytree` and `pytree_metadata`. | Medium | 4/14/2026 |
| v0.11.36 | ### Added - #v1 Add auto loading functionality to `load_pytree` and `pytree_metadata`. | Medium | 4/14/2026 |
| v0.11.36 | ### Added - #v1 Add auto loading functionality to `load_pytree` and `pytree_metadata`. | Medium | 4/14/2026 |
| v0.11.36 | ### Added - #v1 Add auto loading functionality to `load_pytree` and `pytree_metadata`. | Medium | 4/14/2026 |
| v0.11.36 | ### Added - #v1 Add auto loading functionality to `load_pytree` and `pytree_metadata`. | Medium | 4/14/2026 |
| v0.11.36 | ### Added - #v1 Add auto loading functionality to `load_pytree` and `pytree_metadata`. | Medium | 4/14/2026 |
| v0.11.36 | ### Added - #v1 Add auto loading functionality to `load_pytree` and `pytree_metadata`. | Medium | 4/14/2026 |
| v0.11.36 | ### Added - #v1 Add auto loading functionality to `load_pytree` and `pytree_metadata`. | Medium | 4/14/2026 |
| v0.11.36 | ### Added - #v1 Add auto loading functionality to `load_pytree` and `pytree_metadata`. | Medium | 4/14/2026 |
| v0.11.36 | ### Added - #v1 Add auto loading functionality to `load_pytree` and `pytree_metadata`. | Medium | 4/14/2026 |
| v0.11.36 | ### Added - #v1 Add auto loading functionality to `load_pytree` and `pytree_metadata`. | Medium | 4/14/2026 |
| v0.11.36 | ### Added - #v1 Add auto loading functionality to `load_pytree` and `pytree_metadata`. | Medium | 4/14/2026 |
| v0.11.35 | ### Added - Add PyTorch DCP (Distributed Checkpoint) to the benchmark suite, with GCS connector and fsspec options. - Eliminate unadvertised flax dependency from tree/utils. ### Fixed - #v1 Fix import path for `step` in `orbax.checkpoint.experimental.v1.path`. | Medium | 4/13/2026 |
| v0.11.34 | Release v0.11.34 | Medium | 4/8/2026 |
| v0.11.33 | Release v0.11.33 | Low | 2/18/2026 |
| v0.11.32 | Release v0.11.32 | Low | 1/21/2026 |
| v0.11.30 | Release v0.11.30 | Low | 11/27/2025 |
| v0.11.28 | Release v0.11.28 | Low | 11/6/2025 |
| v0.11.27 | Release v0.11.27 | Low | 11/5/2025 |
| v0.11.26 | Release v0.11.26 | Low | 10/22/2025 |
| v0.11.25 | Release v0.11.25 | Low | 9/11/2025 |
| v0.11.24 | Release v0.11.24 | Low | 8/28/2025 |
| v0.11.23 | Release v0.11.23 | Low | 8/18/2025 |
| v0.11.22 | Release v0.11.22 | Low | 8/14/2025 |
| v0.11.21 | Release v0.11.21 | Low | 8/7/2025 |
| v0.11.20 | Release v0.11.20 | Low | 7/28/2025 |
| v0.11.19 | Release v0.11.19 | Low | 7/9/2025 |
| v0.11.18 | Release v0.11.18 | Low | 7/1/2025 |
| v0.11.17 | Release v0.11.17 | Low | 6/30/2025 |
| v0.11.16 | Release v0.11.16 | Low | 6/18/2025 |
| v0.11.15 | Release v0.11.15 | Low | 6/13/2025 |
| v0.11.14 | Release v0.11.14 | Low | 6/10/2025 |
| v0.11.13 | Release v0.11.13 | Low | 5/1/2025 |
| v0.11.12 | Release v0.11.12 | Low | 4/9/2025 |
| v0.11.11 | Release v0.11.11 | Low | 4/7/2025 |
| v0.11.10 | Release v0.11.10 | Low | 3/20/2025 |
| v0.11.9 | Release v0.11.9 | Low | 3/17/2025 |
| v0.11.8 | Release v0.11.8 | Low | 3/7/2025 |
| v0.11.7 | Release v0.11.7 | Low | 3/7/2025 |
| v0.11.6 | Release v0.11.6 | Low | 2/21/2025 |
| v0.11.5 | Release v0.11.5 | Low | 2/11/2025 |
| v0.11.4 | Release v0.11.4 | Low | 2/7/2025 |
| v0.11.3 | Release v0.11.3 | Low | 2/6/2025 |
| v0.11.2 | Release v0.11.2 | Low | 2/4/2025 |
| v0.11.1 | Release v0.11.1 | Low | 1/28/2025 |
| v0.11.0 | Release v0.11.0 | Low | 12/30/2024 |
| v0.10.3 | Release v0.10.3 | Low | 12/19/2024 |
| v0.10.2 | Release v0.10.2 | Low | 12/4/2024 |
| v0.10.1 | Release v0.10.1 | Low | 11/22/2024 |
| v0.10.0 | Release v0.10.0 | Low | 11/22/2024 |
| v0.9.1 | Release v0.9.1 | Low | 11/11/2024 |
| v0.9.0 | Release v0.9.0 | Low | 11/8/2024 |
| v0.8.0 | Release v0.8.0 | Low | 10/29/2024 |
| v0.7.0 | Release v0.7.0 | Low | 10/7/2024 |
| v0.6.4 | Release v0.6.4 | Low | 9/17/2024 |
| v0.6.3 | Release v0.6.3 | Low | 9/10/2024 |
| v0.1.7 | ### Added - Support for OCDBT driver in Tensorstore. | Low | 3/29/2023 |
| v0.1.6 | ### Fixed - Small bug fixes. | Low | 3/22/2023 |
| v0.1.5 | ### Added - Use a more precise timestamp when generating temporary directory names to permit more than one concurrent checkpointing attempt per second. - Automatic import of nest_asyncio. | Low | 3/17/2023 |
| v0.1.4 | ### Added - Support for generic transformation function in PyTreeCheckpointHandler. - Support n-digit checkpoint step format. ### Fixed - Eliminate Flax dependency to fix circular dependency problem. | Low | 3/15/2023 |
| v0.1.3 | ### Added - `sharding` option on `ArrayRestoreArgs | Low | 3/3/2023 |
Dependencies & License Audit
Loading dependencies...
Similar Packages
More from pypi
markitdownUtility tool for converting various files to Markdown
fastapiFastAPI framework, high performance, easy to learn, fast to code, ready for production
djangoA high-level Python web framework that encourages rapid development and clean, pragmatic design.
flaskA simple framework for building complex web applications.
More in Frameworks
langchainThe agent engineering platform
deer-flowAn open-source long-horizon SuperAgent harness that researches, codes, and creates. With the help of sandboxes, memories, tools, skill, subagents and message gateway, it handles different levels of ta
tqdmFast, Extensible Progress Meter
simBuild, deploy, and orchestrate AI agents. Sim is the central intelligence layer for your AI workforce.