Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Aug 6, 2025

• Change the remapping logic for how device ids are mapped across restarts.

  Show description for c6a1902
  

  ishark

  authored and

  Orbax Authors

  committed
  c6a1902

  Copy full SHA for c6a1902

• Fix incorrectly specified timeouts waiting for signals in background threads. Decrease timeout to 60s.

  Show description for 71230ac
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  71230ac

  Copy full SHA for 71230ac

• Make TemporaryPath.finalize async.

  Show description for d580faf
  

  BlaziusMaximus

  authored and

  Orbax Authors

  committed
  d580faf

  Copy full SHA for d580faf

Commits on Aug 5, 2025

• Make TemporaryPath.finalize async.

  Show description for faddc56
  

  BlaziusMaximus

  authored and

  Orbax Authors

  committed
  faddc56

  Copy full SHA for faddc56

• Move v0 metadata loading function under `loading/v0_compatibility.py`. Modify `orbax_layout.py` to allow for metadata handling within the layout itself.

  Show description for 9079ad0
  

  proximityy

  authored and

  Orbax Authors

  committed
  9079ad0

  Copy full SHA for 9079ad0

• Internal change

  Show description for bd4028f
  

  mxberlot

  authored and

  Orbax Authors

  committed
  bd4028f

  Copy full SHA for bd4028f

• Use the path name in the barrier key when creating a new temporary path in the v1 asynchronous checkpoint.

  Show description for cc29b97
  

  Orbax Authors

  committed
  cc29b97

  Copy full SHA for cc29b97

Commits on Aug 4, 2025

• Add options to control number of replicas and min bytes in replica-parallel.

  Show description for bfc521a
  

  Orbax Authors

  committed
  bfc521a

  Copy full SHA for bfc521a

• Internal change

  Show description for 6afa4c8
  

  mxberlot

  authored and

  Orbax Authors

  committed
  6afa4c8

  Copy full SHA for 6afa4c8

Commits on Aug 2, 2025

• Report gbytes metrics with decimal point to see more details in the dashboard

  Show description for a447315
  

  abhishek002002

  authored and

  Orbax Authors

  committed
  a447315

  Copy full SHA for a447315

• Revise PytreeHandler's is_handleable and is_abstract_handleable to validate tree structure only

  Show description for 9314b62
  

  ChromeHearts

  authored and

  Orbax Authors

  committed
  9314b62

  Copy full SHA for 9314b62

Commits on Aug 1, 2025

• Add custom LeafHandler documentation notebook.

  Show description for 203d085
  

  ChromeHearts

  authored and

  Orbax Authors

  committed
  203d085

  Copy full SHA for 203d085

• No public description

  Show description for d371a61
  

  Orbax Authors

  committed
  d371a61

  Copy full SHA for d371a61

• Log `BasePyTreeCheckpointHandler` configuration parameters at INFO level, and include save and restore byte limits in this log.

  Show description for 3e60fe0
  

  dicentra13

  authored and

  Orbax Authors

  committed
  3e60fe0

  Copy full SHA for 3e60fe0

• Provide default xla setting.

  Show description for 68f2c4a
  

  jerryxyj

  authored and

  Orbax Authors

  committed
  68f2c4a

  Copy full SHA for 68f2c4a

• No public description

  Show description for 83c6034
  

  danielsuo

  authored and

  Orbax Authors

  committed
  83c6034

  Copy full SHA for 83c6034

Commits on Jul 31, 2025

• No public description

  Show description for 5c981a2
  

  Orbax Authors

  committed
  5c981a2

  Copy full SHA for 5c981a2

• No public description

  Show description for fd0938e
  

  Orbax Authors

  committed
  fd0938e

  Copy full SHA for fd0938e

• #cleanup De-duplicate scattered implementations of async Path functions and centralize definitions.

  Show description for 6849da1
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  6849da1

  Copy full SHA for 6849da1

• No public description

  Show description for 6606e0e
  

  Orbax Authors

  committed
  6606e0e

  Copy full SHA for 6606e0e

• Add options to control number of replicas and min bytes in replica-parallel.

  Show description for b65659e
  

  niketkumar

  authored and

  Orbax Authors

  committed
  b65659e

  Copy full SHA for b65659e

• Add options to control number of replicas and min bytes in replica-parallel.

  Show description for 5fe9e17
  

  Orbax Authors

  committed
  5fe9e17

  Copy full SHA for 5fe9e17

• Add `ApplyBfloat16OptimizationV2` function in pybinding.

  Show description for 53b7725
  

  jerryxyj

  authored and

  Orbax Authors

  committed
  53b7725

  Copy full SHA for 53b7725

Commits on Jul 30, 2025

• Add `is_saving_in_progress` to `v1.training.Checkpointer`.

  Show description for 1fe0961
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  1fe0961

  Copy full SHA for 1fe0961

• [OBM] Separating expected manifest from obm_export_test to testdata

  Show description for ebf9e64
  

  Orbax Authors

  committed
  ebf9e64

  Copy full SHA for ebf9e64

• Catch snapshot FileNotFoundErrors by default in checkpoint utils.

  Show description for 6688d29
  

  BlaziusMaximus

  authored and

  Orbax Authors

  committed
  6688d29

  Copy full SHA for 6688d29

• Add `metadata()` method for loading with SafeTensors checkpoint format.

  Show description for 91a605c
  

  proximityy

  authored and

  Orbax Authors

  committed
  91a605c

  Copy full SHA for 91a605c

• Fix the copybara issue

  Show description for ef7ad24
  

  jerryxyj

  authored and

  Orbax Authors

  committed
  ef7ad24

  Copy full SHA for ef7ad24

• Release 0.0.7. of OBM

  Show description for a350f07
  

  Orbax Authors

  committed
  a350f07

  Copy full SHA for a350f07

Commits on Jul 29, 2025

• No public description

  Show description for 2a943ca
  

  jerryxyj

  authored and

  Orbax Authors

  committed
  2a943ca

  Copy full SHA for 2a943ca

• optimize gcs interaction by avoiding list-object calls

  Show description for d00682c
  

  Orbax Authors

  committed
  d00682c

  Copy full SHA for d00682c

• Add model surgery api preview to docs guide.

  Show description for 373ed05
  

  BlaziusMaximus

  authored and

  Orbax Authors

  committed
  373ed05

  Copy full SHA for 373ed05

• #cleanup Move `record_saved_duration` to `atomicity.py`.

  Show description for de86be0
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  de86be0

  Copy full SHA for de86be0

• Fix broken links in checkpoint announcements document

  Show description for 324dc0b
  

  Orbax Authors

  committed
  324dc0b

  Copy full SHA for 324dc0b

• Internal change.

  Show description for 9397716
  

  BlaziusMaximus

  authored and

  Orbax Authors

  committed
  9397716

  Copy full SHA for 9397716

Pagination

Previous

Next