Skip to content

Pull requests: google/orbax

New pull request New
320 Open 2,446 Closed
320 Open 2,446 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

Update build workflow schedule and triggers.
#2931 opened Feb 23, 2026 by copybara-service bot Loading…
1
Internal change
#2930 opened Feb 23, 2026 by copybara-service bot Loading…
Enable export-time serialization of XLA GPU flags for Orbax models to natively expose memory optimization features like Latency Hiding Scheduler and Shard Autotuning.
#2928 opened Feb 23, 2026 by copybara-service bot Loading…
1
Add benchmarks for P2P CheckpointManager.
#2927 opened Feb 23, 2026 by copybara-service bot Loading…
1
Restore and broadcast benchmark.
#2925 opened Feb 21, 2026 by copybara-service bot Loading…
Internal change
#2924 opened Feb 20, 2026 by copybara-service bot Loading…
1
Internal change.
#2923 opened Feb 20, 2026 by copybara-service bot Loading…
Internal change
#2922 opened Feb 20, 2026 by copybara-service bot Loading…
1
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…
1
(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…
Internal change
#2915 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…
1
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…
#v1 Add a fresh implementation of NumpyLeafHandler to reduce unnecessary delegation complexity (the actual implementation requires about the same LOC as delegations, and is not expected to change).
#2909 opened Feb 18, 2026 by copybara-service bot Loading…
Internal change
#2905 opened Feb 18, 2026 by copybara-service bot Loading…
Internal change.
#2903 opened Feb 18, 2026 by copybara-service bot Loading…
1
Add a GitHub Actions job to run Orbax benchmarks using xpk.
#2895 opened Feb 17, 2026 by copybara-service bot Loading…
1
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…
Add escape hatch for jax_init_info\
#2888 opened Feb 13, 2026 by copybara-service bot Loading…
1
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
ProTip! Type g i on any issue or pull request to go back to the issue listing page.