-
Notifications
You must be signed in to change notification settings - Fork 81
Pull requests: google/orbax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
#p2p Cleanup P2P restore directories after use
#2861
opened Feb 7, 2026 by
copybara-service
bot
Loading…
Add flags for PyTorch & DCP support in the Orbax checkpoint benchmark launcher.
#2859
opened Feb 5, 2026 by
copybara-service
bot
Loading…
#p2p Enhance P2P restore with coordinated fallback and global step resolution.
#2855
opened Feb 5, 2026 by
copybara-service
bot
Loading…
Refactor composite load and save functionality into orbax layout
#2854
opened Feb 5, 2026 by
copybara-service
bot
Loading…
Just Update the README.md, CONTRIBUTING.md and CHANGELOG.md
#2848
opened Feb 4, 2026 by
lshariprasad
Loading…
[Orbax] Optimize safetensors on-device loading by using a host RAM buffer.
#2846
opened Feb 4, 2026 by
copybara-service
bot
Loading…
Add MANIFEST.in to include proto and pb2 files in Orbax Export.
#2844
opened Feb 3, 2026 by
copybara-service
bot
Loading…
#p2p Add Grain data iterator checkpointing to P2P
#2836
opened Feb 1, 2026 by
copybara-service
bot
Loading…
Update benchmark utility to support replicated arrays
#2830
opened Jan 28, 2026 by
copybara-service
bot
Loading…
#v1 Clarify blocking behavior if a save is already ongoing for
Checkpointer.
#2828
opened Jan 28, 2026 by
copybara-service
bot
Loading…
Update Copybara to rewrite xla_data.proto import path.
#2823
opened Jan 27, 2026 by
copybara-service
bot
Loading…
Add a V1 benchmark in the new benchmarking setup.
#2819
opened Jan 27, 2026 by
copybara-service
bot
Loading…
Add support for loading graphs from pipeline components.
#2809
opened Jan 23, 2026 by
copybara-service
bot
Loading…
Previous Next
ProTip!
Type g i on any issue or pull request to go back to the issue listing page.