Skip to content

Pull requests: google/orbax

New pull request New
328 Open 2,510 Closed
328 Open 2,510 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

Add DeletionOptions to Orbax v1 Context.
#3006 opened Mar 20, 2026 by copybara-service bot Loading…
Fix a couple of dtype mappings in safetensor loading.
#3005 opened Mar 19, 2026 by copybara-service bot Loading…
1
Internal change.
#3000 opened Mar 17, 2026 by copybara-service bot Loading…
Continuous checkpointing is disabled, for experiments that store data in CNS, when storage and compute are not located within the same cluster and metro.
#2998 opened Mar 17, 2026 by copybara-service bot Loading…
1
Add overall timeout in v1 PyTreeHandler.
#2997 opened Mar 14, 2026 by copybara-service bot Loading…
Provide different storage options in PyTorch checkpoint benchmarks.
#2993 opened Mar 12, 2026 by copybara-service bot Loading…
Correct logging imports
#2992 opened Mar 12, 2026 by Michael-T-McCann Loading…
1
1
Internal change.
#2989 opened Mar 11, 2026 by copybara-service bot Loading…
Internal change
#2986 opened Mar 11, 2026 by copybara-service bot Loading…
Internal change
#2985 opened Mar 11, 2026 by copybara-service bot Loading…
Refactor LeafHandlerRegistry state management and handler resolution logic.
#2977 opened Mar 9, 2026 by copybara-service bot Loading…
Cloud Run OSS IT
#2973 opened Mar 9, 2026 by copybara-service bot Loading…
Support the XLA GPU compilation flags in Orbax
#2971 opened Mar 6, 2026 by copybara-service bot Loading…
1
Perf testing P2P
#2968 opened Mar 5, 2026 by copybara-service bot Loading…
1
Add Auto-Assignment GitHub Action for Issues
#2967 opened Mar 5, 2026 by Varun-S10 Loading…
Separate Pathways tests in Orbax OSS build.
#2957 opened Feb 28, 2026 by copybara-service bot Loading…
1
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…
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…
Improve Safetensors loading performance with file pooling and parallelization.
#2935 opened Feb 24, 2026 by copybara-service bot Loading…
Add benchmarks for P2P CheckpointManager.
#2927 opened Feb 23, 2026 by copybara-service bot Loading…
1
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.