-
Notifications
You must be signed in to change notification settings - Fork 82
Pull requests: google/orbax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Update build workflow schedule and triggers.
#2931
opened Feb 23, 2026 by
copybara-service
bot
Loading…
Add benchmarks for P2P CheckpointManager.
#2927
opened Feb 23, 2026 by
copybara-service
bot
Loading…
Add a resharding benchmark. This benchmark only loads the checkpoint repeatedly. The source checkpoint is expected to have been generated on a different topology and/or different sharding. It relies on a sharding config file to dictate the new shardings for the loaded checkpoint.
#2921
opened Feb 20, 2026 by
copybara-service
bot
Loading…
Enable multiprocess testing for Orbax serialization tests.
#2919
opened Feb 20, 2026 by
copybara-service
bot
Loading…
(Experimental) Simple prototype testing for async_save_test: concurrent async saves, interleaved async save operations, testing failure behavior.
#2918
opened Feb 19, 2026 by
copybara-service
bot
Loading…
Re-enable multiprocess unit tests using a new multihost setup script.
#2913
opened Feb 19, 2026 by
copybara-service
bot
Loading…
Add generated v0 and v1 static checkpoints to testing directory third_party/py/orbax/checkpoint/experimental/v1/_src/testing/compatibility
#2912
opened Feb 19, 2026 by
copybara-service
bot
Loading…
Create scripts that will generate static v0 and v1 checkpoints for backwards compatibility testing
#2911
opened Feb 19, 2026 by
copybara-service
bot
Loading…
Add a GitHub Actions job to run Orbax benchmarks using xpk.
#2895
opened Feb 17, 2026 by
copybara-service
bot
Loading…
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…
Previous Next
ProTip!
Type g i on any issue or pull request to go back to the issue listing page.