-
Notifications
You must be signed in to change notification settings - Fork 98
Pull requests: google/orbax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
#v1 Add
metadata method in CheckpointLayout mirroring top-level API.
#3362
opened Jun 12, 2026 by
copybara-service
Bot
Loading…
Open source emergency checkpointing tests and fix OSS JAX simulation compatibility.
#3359
opened Jun 12, 2026 by
copybara-service
Bot
Loading…
Improve cancellation handling in Orbax Checkpoint V1.
#3358
opened Jun 12, 2026 by
copybara-service
Bot
Loading…
use public path for
tree_util.flatten_one_level
#3354
opened Jun 11, 2026 by
copybara-service
Bot
Loading…
Adds /checkpoint/testing/local_path_test.py to the presubmit tests.
#3352
opened Jun 11, 2026 by
copybara-service
Bot
Loading…
Fork CloudPathwaysArrayHandler into Orbax from pathwaysutils.
#3351
opened Jun 11, 2026 by
copybara-service
Bot
Loading…
Update expected error message in save_load_test.py.
#3350
opened Jun 11, 2026 by
copybara-service
Bot
Loading…
Adds /checkpoint/experimental/v1/_src/synchronization/multihost_test.py to the presubmit tests.
#3349
opened Jun 11, 2026 by
copybara-service
Bot
Loading…
Allow saving of empty containers and custom pytrees.
#3346
opened Jun 10, 2026 by
copybara-service
Bot
Loading…
Adds
checkpoint_manager_slice_test.py in Copybara exports to the presubmit tests.
#3339
opened Jun 10, 2026 by
copybara-service
Bot
Loading…
Synchronize OperationIdGenerator across processes in multihost tests.
#3338
opened Jun 10, 2026 by
copybara-service
Bot
Loading…
Add a usage guide for the checkpoint benchmark framework
pull ready
Ready to be pulled from GitHub into Google
#3333
opened Jun 9, 2026 by
mridul-sahu
Collaborator
Loading…
3 of 8 tasks
Track read storage type metric on checkpoint load and restore operations.
#3322
opened Jun 6, 2026 by
copybara-service
Bot
Loading…
Previous Next
ProTip!
Follow long discussions with comments:>50.