Skip to content

Releases: google/orbax

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

v0.11.26

22 Oct 15:32

Choose a tag to compare

Release 0.11.26

PiperOrigin-RevId: 822601209

v0.11.25

11 Sep 17:40

Choose a tag to compare

Bump minimum JAX version to v0.6.0

PiperOrigin-RevId: 805898990