# orbax-checkpoint

> Orbax Checkpoint

- **URL**: https://www.freshcrate.ai/projects/orbax-checkpoint
- **Author**: pypi
- **Category**: Frameworks
- **Latest version**: `v0.12.0` (2026-06-02)
- **License**: Unknown
- **Source**: http://github.com/google/orbax
- **Homepage**: https://pypi.org/project/orbax-checkpoint/
- **Language**: Python
- **GitHub**: 501 stars, 91 forks
- **Registry**: pypi (`orbax-checkpoint`)
- **Tags**: `checkpoint`, `jax`, `learning`, `machine`, `pypi`, `training`

## 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.

## Recent releases

| Version | Date | Urgency | Changes |
| --- | --- | --- | --- |
| `v0.12.0` | 2026-06-02 | High | ### 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 |
| `v0.11.40` | 2026-05-19 | High | ### 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.) |
| `v0.11.39` | 2026-05-06 | High | ### Added  - Tensorstore non-atomic file I/O locking on local/POSIX-compatible filesystems to avoid unnecessary renames. |
| `v0.11.37` | 2026-05-05 | High | ### 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 |
| `0.11.36` | 2026-04-21 | Low | Imported from PyPI (0.11.36) |
| `v0.11.36` | 2026-04-14 | Medium | ### Added  - #v1 Add auto loading functionality to `load_pytree` and `pytree_metadata`. |
| `v0.11.36` | 2026-04-14 | Medium | ### Added  - #v1 Add auto loading functionality to `load_pytree` and `pytree_metadata`. |
| `v0.11.36` | 2026-04-14 | Medium | ### Added  - #v1 Add auto loading functionality to `load_pytree` and `pytree_metadata`. |
| `v0.11.36` | 2026-04-14 | Medium | ### Added  - #v1 Add auto loading functionality to `load_pytree` and `pytree_metadata`. |
| `v0.11.36` | 2026-04-14 | Medium | ### Added  - #v1 Add auto loading functionality to `load_pytree` and `pytree_metadata`. |

## Citation

- HTML: https://www.freshcrate.ai/projects/orbax-checkpoint
- Markdown: https://www.freshcrate.ai/projects/orbax-checkpoint.md
- Dependencies JSON: https://www.freshcrate.ai/api/projects/orbax-checkpoint/deps

_Generated by freshcrate.ai. Indexes pypi releases for AI-agent ecosystem packages._