-
Notifications
You must be signed in to change notification settings - Fork 81
Pull requests: google/orbax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
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…
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…
#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…
Add flags for PyTorch & DCP support in the Orbax checkpoint benchmark launcher.
#2859
opened Feb 5, 2026 by
copybara-service
bot
Loading…
Just Update the README.md, CONTRIBUTING.md and CHANGELOG.md
#2848
opened Feb 4, 2026 by
lshariprasad
Loading…
Previous Next
ProTip!
Type g p on any issue or pull request to go back to the pull request listing page.