Skip to content

Releases: google/orbax

v0.11.39

06 May 19:19

Choose a tag to compare

Added

  • Tensorstore non-atomic file I/O locking on local/POSIX-compatible filesystems
    to avoid unnecessary renames.

Release v0.11.37

05 May 14:56

Choose a tag to compare

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_checkpointing and use local checkpoint engine

Changed

  • Reduced default OCDBT target file size from 2GiB to 400MiB for GCS paths.

v0.11.36

14 Apr 17:03

Choose a tag to compare

Added

  • #v1 Add auto loading functionality to load_pytree and pytree_metadata.

v0.11.35

13 Apr 19:29

Choose a tag to compare

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.

v0.11.34

08 Apr 00:34

Choose a tag to compare

Increment Orbax version.

PiperOrigin-RevId: 896170688

v0.11.33

18 Feb 04:22

Choose a tag to compare

Increment Orbax version.

PiperOrigin-RevId: 871625440

v0.11.32

21 Jan 18:06

Choose a tag to compare

directory compatibility in safetensors_layout

PiperOrigin-RevId: 858993617

v0.11.30

27 Nov 07:58

Choose a tag to compare

No public description

PiperOrigin-RevId: 837257237

v0.11.28

06 Nov 20:43

Choose a tag to compare

#v1 Add `Checkpointer.wait` method.

PiperOrigin-RevId: 829067123

v0.11.27

05 Nov 16:13

Choose a tag to compare

Release 0.11.27

PiperOrigin-RevId: 828475045