-
Notifications
You must be signed in to change notification settings - Fork 76
Pull requests: google/orbax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Refactor
CheckpointLayout splitting load() into load_pytree() and load_checkpointables() each with their own dedicated loading logic.
#2764
opened Jan 9, 2026 by
copybara-service
bot
Loading…
Extract JAX model input signature calculation.
#2762
opened Jan 9, 2026 by
copybara-service
bot
Loading…
Improve multislice zero buffer creation during broadcasting.
#2758
opened Jan 8, 2026 by
copybara-service
bot
Loading…
Refactor CheckpointLayout to be stateless with respect to the path.
#2757
opened Jan 8, 2026 by
copybara-service
bot
Loading…
Refactor CheckpointLayout to be stateless with respect to the path.
#2756
opened Jan 8, 2026 by
copybara-service
bot
Loading…
Default cluster name and improve command output in XPK launch script.
#2755
opened Jan 8, 2026 by
copybara-service
bot
Loading…
Fix absolute link references to use relative path links.
#2748
opened Jan 6, 2026 by
copybara-service
bot
Loading…
Add pytree_checkpoint_handler_test.py to multiprocess tests.
#2745
opened Jan 6, 2026 by
copybara-service
bot
Loading…
Add
save method to CheckpointLayout interface and implementations.
#2743
opened Jan 5, 2026 by
copybara-service
bot
Loading…
Add network metrics monitor in benchmark library.
#2740
opened Dec 23, 2025 by
copybara-service
bot
Loading…
Move Orbax checkpoint benchmark configs to a new google/correctness subdirectory.
#2731
opened Dec 18, 2025 by
copybara-service
bot
Loading…
Add validation for restoring with target trees containing none leaf types
#2727
opened Dec 17, 2025 by
copybara-service
bot
Loading…
#v1 Add a fresh implementation of
StringLeafHandler to reduce unnecessary delegation complexity. Also introduce a typestr function for LeafHandler.
#2713
opened Dec 16, 2025 by
copybara-service
bot
Loading…
Previous Next
ProTip!
Follow long discussions with comments:>50.