Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Jun 16, 2025

• #v1 Fix checkpointer_test flakiness due to iteration over files being concurrently modified by TensorStore.

  Show description for ab324b2
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  ab324b2

  Copy full SHA for ab324b2

Commits on Jun 13, 2025

• Add partial restore omission to v1 PyTreeHandler.

  Show description for cde6bb0
  

  BlaziusMaximus

  authored and

  Orbax Authors

  committed
  cde6bb0

  Copy full SHA for cde6bb0

• Release 0.11.15.

  Show description for 50c4f0b
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  50c4f0b

  Copy full SHA for 50c4f0b

• Added docstring to `keep_period` for `CheckpointManagerOptions`

  Show description for 56f6135
  

  Orbax Authors

  committed
  56f6135

  Copy full SHA for 56f6135

• Fix Continuous Save Decision Policy.

  Show description for 6d57e40
  

  mridul-sahu

  authored and

  Orbax Authors

  committed
  6d57e40

  Copy full SHA for 6d57e40

• Enable keep_every_n GCS checkpoints in emergency checkpoint manager

  Show description for e7e9326
  

  Orbax Authors

  committed
  e7e9326

  Copy full SHA for e7e9326

Commits on Jun 12, 2025

• Remove unnecessary array_metadata_store parameters from PyTreeHandler tests.

  Show description for 8bb6fff
  

  BlaziusMaximus

  authored and

  Orbax Authors

  committed
  8bb6fff

  Copy full SHA for 8bb6fff

• Populate metrics when getting StepMetadata in the CheckpointManager.

  Show description for b892468
  

  BlaziusMaximus

  authored and

  Orbax Authors

  committed
  b892468

  Copy full SHA for b892468

• Support `keep_checkpoints_without_metrics` in the `BestN` preservation policy.

  Show description for 5225171
  

  Orbax Authors

  committed
  5225171

  Copy full SHA for 5225171

• #emergency-checkpoint Stop including barrier processes in barrier names. This information should be redundant, since the barrier ID always corresponds 1:1 with the provided processes ('local_all_st…

  Show description for ae49fe3
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  ae49fe3

  Copy full SHA for ae49fe3

Commits on Jun 11, 2025

• Reduce logging spam in CheckpointManager.metrics().

  Show description for 2177519
  

  BlaziusMaximus

  authored and

  Orbax Authors

  committed
  2177519

  Copy full SHA for 2177519

• #v1 Fix Orbax usage in colab by applying nest_asyncio.

  Show description for 480d561
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  480d561

  Copy full SHA for 480d561

Commits on Jun 10, 2025

• #v1 Use an implementation of `save` independent of `v0.Checkpointer`.

  Show description for f49c222
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  f49c222

  Copy full SHA for f49c222

• Remove all build files from Orbax.

  Show description for 4202a88
  

  mridul-sahu

  authored and

  Orbax Authors

  committed
  4202a88

  Copy full SHA for 4202a88

• Add a ScalarLeafHandler

  Show description for 7f39fc0
  

  ChromeHearts

  authored and

  Orbax Authors

  committed
  7f39fc0

  Copy full SHA for 7f39fc0

• Expand partial restore omission test to include non-sharded arrays.

  Show description for ca14492
  

  BlaziusMaximus

  authored and

  Orbax Authors

  committed
  ca14492

  Copy full SHA for ca14492

• Internal

  Show description for 6f60838
  

  Orbax Authors

  committed
  6f60838

  Copy full SHA for 6f60838

• Increase Orbax version.

  Show description for e50dd5b
  

  mridul-sahu

  authored and

  Orbax Authors

  committed
  e50dd5b

  Copy full SHA for e50dd5b

Commits on Jun 9, 2025

• No public description

  Show description for 069df58
  

  Orbax Authors

  committed
  069df58

  Copy full SHA for 069df58

• Report `/jax/orbax/write/storage_type` metric only from the primary host.

  Show description for fe0920b
  

  dicentra13

  authored and

  Orbax Authors

  committed
  fe0920b

  Copy full SHA for fe0920b

Commits on Jun 6, 2025

• Internal change.

  Show description for ab09f70
  

  dicentra13

  authored and

  Orbax Authors

  committed
  ab09f70

  Copy full SHA for ab09f70

Commits on Jun 4, 2025

• Internal change.

  Show description for 77b1eca
  

  BlaziusMaximus

  authored and

  Orbax Authors

  committed
  77b1eca

  Copy full SHA for 77b1eca

• #v1 Document migration from v0 CheckpointManager to v1 Checkpointer.

  Show description for 64eef8a
  

  niketkumar

  authored and

  Orbax Authors

  committed
  64eef8a

  Copy full SHA for 64eef8a

• Add NumpyLeafhandler to v1 compatible type handler registry.

  Show description for 2e59e41
  

  ChromeHearts

  authored and

  Orbax Authors

  committed
  2e59e41

  Copy full SHA for 2e59e41

Commits on Jun 3, 2025

• Extend metadata test to include fresh handler access.

  Show description for 6963406
  

  BlaziusMaximus

  authored and

  Orbax Authors

  committed
  6963406

  Copy full SHA for 6963406

• Add V1 numpy_leaf_hander implementation.

  Show description for 8b1c2dc
  

  ChromeHearts

  authored and

  Orbax Authors

  committed
  8b1c2dc

  Copy full SHA for 8b1c2dc

• internal monitoring and hashing change

  Show description for 24cc0b2
  

  Orbax Authors

  committed
  24cc0b2

  Copy full SHA for 24cc0b2

• No public description

  Show description for ff5f495
  

  mridul-sahu

  authored and

  Orbax Authors

  committed
  ff5f495

  Copy full SHA for ff5f495

• Disable Async Directory Creation for Local Checkpoint Mangers.

  Show description for 7abc2ab
  

  mridul-sahu

  authored and

  Orbax Authors

  committed
  7abc2ab

  Copy full SHA for 7abc2ab

Commits on Jun 2, 2025

• Enable ObmModule to support multiple apply functions.

  Show description for 32987c2
  

  jerryxyj

  authored and

  Orbax Authors

  committed
  32987c2

  Copy full SHA for 32987c2

• No public description

  Show description for ff412ea
  

  mridul-sahu

  authored and

  Orbax Authors

  committed
  ff412ea

  Copy full SHA for ff412ea

• Removed atomic operations from mesh_consistency.save_process_metadata

  Show description for 0b2b2d8
  

  Orbax Authors

  committed
  0b2b2d8

  Copy full SHA for 0b2b2d8

Commits on May 30, 2025

• No public description

  Show description for a4cc417
  

  Orbax Authors

  committed
  a4cc417

  Copy full SHA for a4cc417

• Simplify `__str__` of NameFormats.

  Show description for 5bfc17f
  

  niketkumar

  authored and

  Orbax Authors

  committed
  5bfc17f

  Copy full SHA for 5bfc17f

Commits on May 29, 2025

• Add a partial restore omission test for Checkpointer.

  Show description for ac226bd
  

  BlaziusMaximus

  authored and

  Orbax Authors

  committed
  ac226bd

  Copy full SHA for ac226bd

Pagination

Previous

Next