-
Notifications
You must be signed in to change notification settings - Fork 85
Pull requests: google/orbax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
#p2p Propagate sharding to allow restore with resize
#2959
opened Mar 2, 2026 by
copybara-service
bot
Loading…
Separate Pathways tests in Orbax OSS build.
#2957
opened Feb 28, 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…
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…
Add Multi-tiered Checkpointing Support to Pathways Single Controller
#2937
opened Feb 24, 2026 by
copybara-service
bot
Loading…
Improve Safetensors loading performance with file pooling and parallelization.
#2935
opened Feb 24, 2026 by
copybara-service
bot
Loading…
Support the XLA GPU compilation flags in Orbax
#2928
opened Feb 23, 2026 by
copybara-service
bot
Loading…
Add benchmarks for P2P CheckpointManager.
#2927
opened Feb 23, 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.