Skip to content

Pull requests: google/orbax

New pull request New
312 Open 2,406 Closed
312 Open 2,406 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

Delete resolution.py and its build rule.
#2882 opened Feb 12, 2026 by copybara-service bot Loading…
Relocate handler resolution logic to orbax_layout.py
#2881 opened Feb 12, 2026 by copybara-service bot Loading…
Internal change.
#2880 opened Feb 11, 2026 by copybara-service bot Loading…
Internal change
#2879 opened Feb 11, 2026 by copybara-service bot Loading…
1
#v1 Refactor OrbaxLayout metadata retrieval to use internal logic, removing dependency on v0 checkpointer.
#2878 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…
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
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…
#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
my public
#2835 opened Jan 30, 2026 by copybara-service bot Loading…
ProTip! Mix and match filters to narrow down what you’re looking for.