Skip to content

Pull requests: google/orbax

New pull request New
372 Open 2,716 Closed
372 Open 2,716 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 Add stringent validation for state abstract and concrete leaf types.
#3273 opened May 29, 2026 by copybara-service Bot Loading…
No public description
#3272 opened May 29, 2026 by copybara-service Bot Loading…
Modernize OSS linting stack: replace Pylint, Pyink, and Isort with Ruff
#3271 opened May 29, 2026 by copybara-service Bot Loading…
Add validation for unique component weights_name in ObmExport.
#3270 opened May 29, 2026 by copybara-service Bot Loading…
Internal Changes
#3269 opened May 29, 2026 by copybara-service Bot Loading…
#v1 Add a test for saving and loading grain iterator.
#3267 opened May 28, 2026 by copybara-service Bot Loading…
No public description
#3261 opened May 28, 2026 by copybara-service Bot Loading…
1
Refactor tensor handling and add dynamic SafeTensors loading.
#3259 opened May 27, 2026 by copybara-service Bot Loading…
Add Orbax v1 Context User Guide.
#3249 opened May 26, 2026 by copybara-service Bot Loading…
Include v1 save_load_test in OSS.
#3243 opened May 25, 2026 by copybara-service Bot Loading…
No public description
#3236 opened May 22, 2026 by copybara-service Bot Loading…
1
Internal
#3234 opened May 21, 2026 by copybara-service Bot Loading…
Modularize PyTree structure walk and alignment logic.
#3233 opened May 21, 2026 by copybara-service Bot Loading…
add save & restore benchmark configuration files.
#3232 opened May 21, 2026 by copybara-service Bot Loading…
Security: Thread Safety Issue in Asyncio Utils
#3224 opened May 20, 2026 by tomaioo Loading…
1
Adjust renamed Orbax V1 API calls.
#3220 opened May 19, 2026 by copybara-service Bot Loading…
No public description
#3212 opened May 18, 2026 by copybara-service Bot Loading…
1
Modularize BasePyTreeCheckpointHandler metadata persistence logic into MetadataManager.
#3209 opened May 15, 2026 by copybara-service Bot Loading…
Modularize BasePyTreeCheckpointHandler concurrency and telemetry logic into AsyncIoEngine.
#3205 opened May 14, 2026 by copybara-service Bot Loading…
#v1 Refactor Context usage to reflect new namespace pattern.
#3203 opened May 14, 2026 by copybara-service Bot Loading…
Redesign partial saving to use pending directories.
#3194 opened May 13, 2026 by copybara-service Bot Loading…
Introduce TransferTracker for Orbax Checkpoint D2H transfers.
#3192 opened May 13, 2026 by copybara-service Bot Loading…
Generate static checkpoint managed checkpoints using generate_v0_checkpoint_manager_checkpoints and generate_v1_checkpointer_checkpoints scripts using the following commands:
#3169 opened May 7, 2026 by copybara-service Bot Loading…
Add v0 checkpoint manager and v1 checkpointer checkpoint generation scripts.
#3168 opened May 7, 2026 by copybara-service Bot Loading…
[monitoring] Pass config-managed key-value tags to monitoring listeners.
#3153 opened May 4, 2026 by copybara-service Bot Loading…
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.