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 |
|---|---|---|---|
| 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.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
waybackpyPython package that interfaces with the Internet Archive's Wayback Machine APIs. Archive pages and retrieve archived pages easily.3.0.6