Skip to content

Pull requests: google/orbax

New pull request New
327 Open 2,464 Closed
327 Open 2,464 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

Separate Pathways tests in Orbax OSS build.
#2957 opened Feb 28, 2026 by copybara-service bot Loading…
1
#v1 Shorten naming for publicly-exposed save_decision_policies and preservation_policies.
#2956 opened Feb 27, 2026 by copybara-service bot Loading…
#p2p Implement CheckpointManager.reload() properly
#2954 opened Feb 27, 2026 by copybara-service bot Loading…
1
Internal change
#2953 opened Feb 27, 2026 by copybara-service bot Loading…
Internal change
#2952 opened Feb 26, 2026 by copybara-service bot Loading…
Internal change
#2951 opened Feb 26, 2026 by copybara-service bot Loading…
Create v1 ocp.pytree_metadata backwards compatibility tests against static v0 and v1 checkpoints.
#2950 opened Feb 26, 2026 by copybara-service bot Loading…
Create v1 ocp.load_checkpointables backwards compatibility tests against static v0 and v1 checkpoints.
#2949 opened Feb 26, 2026 by copybara-service bot Loading…
Create v1 ocp.load_pytree backwards compatibility tests against static v0 and v1 checkpoints.
#2948 opened Feb 26, 2026 by copybara-service bot Loading…
Internal change.
#2946 opened Feb 25, 2026 by copybara-service bot Loading…
Raise a more useful error when doing a dtensor export with pspecs but not all params are jax arrays.
#2945 opened Feb 25, 2026 by copybara-service bot Loading…
Introduce registered handler to alternate handler typestrs mapping to v1 API.
#2942 opened Feb 24, 2026 by copybara-service bot Loading…
Refine resolution logic to ensure valid handler matching for abstract checkpointables.
#2941 opened Feb 24, 2026 by copybara-service bot Loading…
Adjust layout error logic when user passes incorrect loading path (root/child)
#2940 opened Feb 24, 2026 by copybara-service bot Loading…
Improve SerializationContext and DeserializationContext class docstrings
#2939 opened Feb 24, 2026 by copybara-service bot Loading…
1
Add Multi-tiered Checkpointing Support to Pathways Single Controller
#2937 opened Feb 24, 2026 by copybara-service bot Loading…
Include Orbax Checkpoint experimental/emergency tests in OSS.
#2936 opened Feb 24, 2026 by copybara-service bot Loading…
1
Improve Safetensors loading performance with file pooling and parallelization.
#2935 opened Feb 24, 2026 by copybara-service bot Loading…
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…
(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…
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.