Skip to content

Pull requests: google/orbax

New pull request New
316 Open 2,425 Closed
316 Open 2,425 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

Internal change
#2905 opened Feb 18, 2026 by copybara-service bot Loading…
Disable multiprocess-unit-tests job.
#2904 opened Feb 18, 2026 by copybara-service bot Loading…
1
Internal change.
#2903 opened Feb 18, 2026 by copybara-service bot Loading…
1
Internal change.
#2902 opened Feb 18, 2026 by copybara-service bot Loading…
1
Internal change.
#2901 opened Feb 18, 2026 by copybara-service bot Loading…
1
Use device.memory_stats for accelerator HBM size.
#2900 opened Feb 18, 2026 by copybara-service bot Loading…
1
#p2p Specify all processes in multiprocess option for local
#2898 opened Feb 17, 2026 by copybara-service bot Loading…
1
#p2p Implement latest_step correctly
#2897 opened Feb 17, 2026 by copybara-service bot Loading…
1
#p2p Raise FileNotFoundError for not finding steps anywhere
#2896 opened Feb 17, 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…
#p2p Enable SingleReplicaRestore for PersistentCheckpointManager
#2891 opened Feb 13, 2026 by copybara-service bot Loading…
1
#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
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…
Update multiprocess unit tests to use an 8-TPU runner.
#2872 opened Feb 10, 2026 by copybara-service bot Loading…
1
#v1 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…
Ensure colocated CPU sharding uses 'pinned_host' memory kind.
#2865 opened Feb 9, 2026 by copybara-service bot Loading…
1
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.