Releases: google/orbax
Releases · google/orbax
v0.11.39
Release v0.11.37
Added
- Add colocated runtime helpers for Pathways MTC.
- #v1 Centralize
StorageOptionsintoArrayOptionsand 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
LeafHandleras aCheckpointableHandler, so that ordinary PyTree
leaves can also be saved as individual checkpointables. - Move MTC files to multi_tier_checkpointing and use local checkpoint engine
Changed
- Reduced default OCDBT target file size from 2GiB to 400MiB for GCS paths.