Skip to main content

Ctrl+K

Orbax documentation

Getting Started

Installation
Introduction to Checkpointing with Orbax
Working with Checkpointables
Working with PyTree Checkpoints
Checkpointing in a Training Loop

Performance

Asynchronous Checkpointing
Checkpoint Format Guide

Advanced Usage

Customizing Checkpointing Behavior
Configuring Specialized Features
Partial Saving
Model Surgery

API Reference

orbax.checkpoint V1 API Reference

Migrating to Orbax V1

What is the Orbax V1 API?
Orbax v0 to v1 Migration Guide
Legacy API (V0)

Exporting

Exporting with Orbax
Checkpointing and Exporting JAX Models: An End-to-End Guide with Orbax
orbax.export API Reference

Additional Information

Contributors

.rst

Checkpoint Deleter

Checkpoint Deleter#

Checkpoint deleter.

previous

Step entities

next

Preservation Policy

By Orbax Contributors

© Copyright 2026, Google.