Skip to content

Pull requests: google/orbax

New pull request New
310 Open 2,391 Closed
310 Open 2,391 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Ensure colocated CPU sharding uses 'pinned_host' memory kind.
#2865 opened Feb 9, 2026 by copybara-service bot Loading…
1
Refactor multiprocess test execution in Orbax.
#2864 opened Feb 9, 2026 by copybara-service bot Loading…
1
Internal
#2863 opened Feb 9, 2026 by copybara-service bot Loading…
1
#p2p Make Composite immutable
#2862 opened Feb 7, 2026 by copybara-service bot Loading…
1
#p2p Cleanup P2P restore directories after use
#2861 opened Feb 7, 2026 by copybara-service bot Loading…
1
Add flags for PyTorch & DCP support in the Orbax checkpoint benchmark launcher.
#2859 opened Feb 5, 2026 by copybara-service bot Loading…
Internal change.
#2857 opened Feb 5, 2026 by copybara-service bot Loading…
Remove obm_to_jax module.
#2856 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…
1
Refactor composite load and save functionality into orbax layout
#2854 opened Feb 5, 2026 by copybara-service bot Loading…
my public
#2849 opened Feb 4, 2026 by copybara-service bot Loading…
Just Update the README.md, CONTRIBUTING.md and CHANGELOG.md
#2848 opened Feb 4, 2026 by lshariprasad Loading…
3
[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…
Testing unit tests
#2843 opened Feb 3, 2026 by copybara-service bot Loading…
1
Migrate python/ directory.
#2842 opened Feb 3, 2026 by copybara-service bot Loading…
1
#p2p Add Grain data iterator checkpointing to P2P
#2836 opened Feb 1, 2026 by copybara-service bot Loading…
1
my public
#2835 opened Jan 30, 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…
Add Safetensors load benchmark.
#2827 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…
Internal change.
#2811 opened Jan 23, 2026 by copybara-service bot Loading…
No public description
#2810 opened Jan 23, 2026 by copybara-service bot Loading…
ProTip! Filter pull requests by the default branch with base:main.