Commits

Branch selector

 6010be5

User selector

All users

Datepicker

All time

Commit History

Commits on Apr 24, 2024

• Refractor Tensorstore-related codes in type_handlers.py

  Show description for 6010be5
  

  ChromeHearts

  authored and

  Orbax Authors

  committed
  6010be5

• Remove OCDBT coordinator-related code.

  Show description for de1a0dc
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  de1a0dc

• internal change

  Show description for e8b144e
  

  Orbax Authors

  committed
  e8b144e

• Internal change.

  Show description for e46cbce
  

  niketkumar

  authored and

  Orbax Authors

  committed
  e46cbce

Commits on Apr 23, 2024

• No public description

  Show description for 512643f
  

  Orbax Authors

  committed
  512643f

• Fix small logging bug.

  Show description for 19b8577
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  19b8577

• Internal change.

  Show description for e158c1f
  

  niketkumar

  authored and

  Orbax Authors

  committed
  e158c1f

• Remove ocdbt_merge and restore_with_serialized_types options from PyTreeCheckpointHandler as they are not in use anywhere.

  Show description for 90c8d85
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  90c8d85

Commits on Apr 22, 2024

• Increase version.

  Show description for dee9991
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  dee9991

• Add Composite logger for Checkpointing Logging

  Show description for 2f2a243
  

  abhinavclemson

  authored and

  Orbax Authors

  committed
  2f2a243

• pass timout_fn() to wait_for_new_checkpoint and _wait_for_new_checkpoint

  Show description for 10f0f27
  

  divyashreepathihalli

  authored and

  Orbax Authors

  committed
  10f0f27

Commits on Apr 20, 2024

• implement all_steps function for experimental emergency CheckpointManager. At the same time, only create sub CheckpointManagers on appropriate slices.

  Show description for f06df1f
  

  Orbax Authors

  committed
  f06df1f

Commits on Apr 17, 2024

• Clarify that CheckpointManager expects itself to be unique across jobs.

  Show description for 4b50229
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  4b50229

• No public description

  Show description for 5b291c3
  

  Orbax Authors

  committed
  5b291c3

• Store jax mesh together with dtensor mesh. This change is a noop.

  Show description for b7ea044
  

  Orbax Authors

  committed
  b7ea044

Commits on Apr 16, 2024

• No public description

  Show description for d904628
  

  Orbax Authors

  committed
  d904628

• Internal change.

  Show description for b8956cd
  

  niketkumar

  authored and

  Orbax Authors

  committed
  b8956cd

Commits on Apr 15, 2024

• Update the comments and test.

  Show description for a174452
  

  maxwillzq

  authored and

  Orbax Authors

  committed
  a174452

Commits on Apr 12, 2024

• Internal change.

  Show description for 4fb9439
  

  niketkumar

  authored and

  Orbax Authors

  committed
  4fb9439

Commits on Apr 11, 2024

• Add logs to indicate when checkpoints are locked and unlocked for evaluation.

  Show description for f4cc33f
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  f4cc33f

Commits on Apr 10, 2024

• For broadcast_one_replica_to_all, delete input arrays as soon as possible to conserve memory. Should result in 2x memory consumption from this function rather than 3x.

  Show description for 5bf26e6
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  5bf26e6

• Internal change.

  Show description for 59c8df3
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  59c8df3

• Add checkpointing_pytrees to the index.

  Show description for 29c6037
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  29c6037

• Fix CHANGELOG duplicate Changed entries.

  Show description for 07658e9
  

  niketkumar

  authored and

  Orbax Authors

  committed
  07658e9

• Add docs on working with PyTrees/arrays. Modify error messages for when sharding is not provided.

  Show description for 470c74c
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  470c74c

• Improve step lookup error message by adding expected names to it.

  Show description for 74adaef
  

  niketkumar

  authored and

  Orbax Authors

  committed
  74adaef

Commits on Apr 9, 2024

• internal code change.

  Show description for 1cf704a
  

  Orbax Authors

  committed
  1cf704a

Commits on Apr 8, 2024

• Create composite step NameFormat.

  Show description for cfbfddf
  

  niketkumar

  authored and

  Orbax Authors

  committed
  cfbfddf

• Add path package to export symbols, also add Step rst docs.

  Show description for 27ac180
  

  niketkumar

  authored and

  Orbax Authors

  committed
  27ac180

Commits on Apr 6, 2024

• Add local type handler registries and refactor PyTreeCheckpointHandler

  Show description for d5cf598
  

  Orbax Authors

  committed
  d5cf598

Commits on Apr 5, 2024

• Internal change.

  Show description for b9593bd
  

  niketkumar

  authored and

  Orbax Authors

  committed
  b9593bd

• Scaffold the Structured Logging in Orbax library.

  Show description for a881027
  

  abhinavclemson

  authored and

  Orbax Authors

  committed
  a881027

Commits on Apr 4, 2024

• In ThreadedCheckpointDeleter, use a daemon thread so that it won't block the main thread during exits. The thread will die when the program completely exits.

  Show description for abad2e4
  

  ChromeHearts

  authored and

  Orbax Authors

  committed
  abad2e4

• internal change

  Show description for 288e736
  

  Orbax Authors

  committed
  288e736

Commits on Apr 3, 2024

• Fix build failures on older Python versions.

  Show description for 5a42887
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  5a42887

Pagination

Previous

Next