Skip to content

Pull requests: google/orbax

New pull request New
314 Open 2,392 Closed
314 Open 2,392 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

#v1 Deprecate and delete CompositeHandler in favor of OrbaxLayout
#2870 opened Feb 10, 2026 by copybara-service bot Loading…
Add Orbax Layout multiprocess tests (forked from composite_handler)
#2869 opened Feb 10, 2026 by copybara-service bot Loading…
In-line CompositeHandler logic into Orbax Layouts
#2868 opened Feb 10, 2026 by copybara-service bot Loading…
Restrict bfloat16 conversion to only float32 types.
#2867 opened Feb 9, 2026 by copybara-service bot Loading…
Add an ability to pass a config file alongside a reference checkpoint path for benchmarking. This allows easily reconstructing an abstract tree.
#2866 opened Feb 9, 2026 by copybara-service bot Loading…
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…
#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…
ProTip! Exclude everything labeled bug with -label:bug.