Skip to content

Pull requests: google/orbax

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

Normalize device kind strings when looking up HBM memory.
#2893 opened Feb 14, 2026 by copybara-service bot Loading…
#v1 Refactor logic for handler resolution and loading checkpointables + additional fallback capabilities for non-standard checkpoint formats.
#2892 opened Feb 13, 2026 by copybara-service bot Loading…
#p2p Enable SingleReplicaRestore for PersistentCheckpointManager
#2891 opened Feb 13, 2026 by copybara-service bot Loading…
1
Internal Change
#2890 opened Feb 13, 2026 by copybara-service bot Loading…
#v1 Add use_load_and_broadcast option.
#2889 opened Feb 13, 2026 by copybara-service bot Loading…
Add escape hatch for jax_init_info\
#2888 opened Feb 13, 2026 by copybara-service bot Loading…
1
Internal change
#2886 opened Feb 13, 2026 by copybara-service bot Loading…
Allow restore+broadcast logic to not require a single-replica sharding parameter, which is always constructed as a sharding over replica-local devices anyway.
#2885 opened Feb 13, 2026 by copybara-service bot Loading…
Fix orbax issue
#2883 opened Feb 12, 2026 by lshariprasad 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 + refactor to read checkpoint metadata externally from get_handlers_from_load.
#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
#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
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…
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
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.