orbax-checkpoint
Orbax Checkpoint
Why this rank:Strong adoptionRelease freshnessHealthy 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.1 | ### Added - Support for `gcs_grpc` driver. - #v1 Add stringent validation for state abstract and concrete leaf types. - Update MTC w/ Pathways to Support Scale Elasticity. | High | 6/24/2026 |
| 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
bamlThe AI framework that adds the engineering to prompt engineering (Python/TS/Ruby/Java/C#/Rust/Go compatible)
saas-builderAI-native SaaS framework that builds full-stack apps using autonomous AI agents
djangoA high-level Python web framework that encourages rapid development and clean, pragmatic design.
sglangSGLang is a fast serving framework for large language models and vision language models.