Skip to main content

Ctrl+K

Orbax documentation

Checkpointing

Announcements
Checkpointing with Orbax
API Overview
Using the Refactored CheckpointManager API
Checkpointing PyTrees of Arrays
Checkpoint Format Guide
Optimized Checkpointing with Tensorstore
Transformations
Preemption Tolerance
Asynchronous Checkpointing
Debugging Guide
orbax.checkpoint API Reference
V1

Exporting

Exporting with Orbax
orbax.export API Reference

Additional Information

Contributors

.rst

model.core.python.polymorphic_function

model.core.python.polymorphic_function#

By Orbax Contributors

© Copyright 2026, Google.