Skip to content

Pull requests: google/orbax

New pull request New
318 Open 2,443 Closed
318 Open 2,443 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

#v1 Fix missing imports and symbols in OSS
#2926 opened Feb 21, 2026 by copybara-service bot Loading…
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…
In benchmarks, modify tensorboard writing to be periodic rather than only at the end.
#2920 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
#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 configs for evaluating resharding under different scenarios.
#2876 opened Feb 10, 2026 by copybara-service bot Loading…
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.