Skip to content

Pull requests: google/orbax

New pull request New
318 Open 2,396 Closed
318 Open 2,396 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

Refactor OrbaxLayout metadata retrieval to use internal logic, removing dependency on v0 checkpointer.
#2878 opened Feb 11, 2026 by copybara-service bot Loading…
Internal only change.
#2877 opened Feb 11, 2026 by copybara-service bot Loading…
Add resharding benchmark.
#2876 opened Feb 10, 2026 by copybara-service bot Loading…
Refactor orbax layout multiprocess test to use OrbaxLayout save and load logic.
#2875 opened Feb 10, 2026 by copybara-service bot Loading…
#v1 Strip down composite handler class and refactor functionality.
#2874 opened Feb 10, 2026 by copybara-service bot Loading…
Internal Changes
#2873 opened Feb 10, 2026 by copybara-service bot Loading…
Update multiprocess unit tests to use an 8-TPU runner.
#2872 opened Feb 10, 2026 by copybara-service bot Loading…
1
Internal change
#2871 opened Feb 10, 2026 by copybara-service bot Loading…
1
#v1 Deprecate and delete CompositeHandler in favor of OrbaxLayout
#2870 opened Feb 10, 2026 by copybara-service bot Loading…
#v1 Add Orbax Layout multiprocess tests (forked from composite_handler)
#2869 opened Feb 10, 2026 by copybara-service bot Loading…
#v1 In-line CompositeHandler logic into Orbax Layouts
#2868 opened Feb 10, 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
#v1 Fork handler resolution functionality from composite_handler into simpler utility file.
#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
ProTip! Adding no:label will show everything without a label.