Skip to content

Pull requests: google/orbax

New pull request New
379 Open 2,760 Closed
379 Open 2,760 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

No public description
#3324 opened Jun 6, 2026 by copybara-service Bot Loading…
Add read storage type metric on checkpoint load and restore operations.
#3323 opened Jun 6, 2026 by copybara-service Bot Loading…
Track read storage type metric on checkpoint load and restore operations.
#3322 opened Jun 6, 2026 by copybara-service Bot Loading…
Optimize the helper function for merging and construct fragments. #MSOSS
#3321 opened Jun 5, 2026 by copybara-service Bot Loading…
Update MTC w/ Pathways to Support Scale Elasticity.
#3320 opened Jun 5, 2026 by copybara-service Bot Loading…
fix(path): fix Windows path compatibility in recursively_copy_files
#3319 opened Jun 5, 2026 by mramansayyad Loading…
3 of 8 tasks
Adds orbax/checkpoint/experimental/v1/_src/partial:saving_multihost_test to the presubmit tests.
#3316 opened Jun 5, 2026 by copybara-service Bot Loading…
Expose helper functions for fragments union and extraction. #MSOSS
#3315 opened Jun 5, 2026 by copybara-service Bot Loading…
Internal change
#3314 opened Jun 5, 2026 by copybara-service Bot Loading…
#v1 Ensure CheckpointLayout method naming is consistent with top-level APIs.
#3313 opened Jun 5, 2026 by copybara-service Bot Loading…
Add support for Orbax metrics in tpu-info.
#3307 opened Jun 3, 2026 by copybara-service Bot Loading…
Add cancellation support to Orbax Futures.
#3303 opened Jun 3, 2026 by copybara-service Bot Loading…
Internal
#3302 opened Jun 2, 2026 by copybara-service Bot Loading…
Modify the telemetry for saving metrics.
#3298 opened Jun 2, 2026 by copybara-service Bot Loading…
ROC: Building model shape attribute and invoking model registration with this enriched model shape data.
#3292 opened Jun 1, 2026 by copybara-service Bot Loading…
1
No public description
#3287 opened Jun 1, 2026 by copybara-service Bot Loading…
No public description
#3285 opened Jun 1, 2026 by copybara-service Bot Loading…
1
#v1 Allow PyTreeCheckpointHandler to save/load bare leaves.
#3284 opened Jun 1, 2026 by copybara-service Bot Loading…
No public description
#3281 opened Jun 1, 2026 by copybara-service Bot Loading…
No public description
#3280 opened Jun 1, 2026 by copybara-service Bot Loading…
Internal Changes
#3269 opened May 29, 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…
Adding model extraction in Orbax write operation for v0 and v1 version.
#3236 opened May 22, 2026 by copybara-service Bot Loading…
1
Modularize PyTree structure walk and alignment logic.
#3233 opened May 21, 2026 by copybara-service Bot Loading…
ProTip! Filter pull requests by the default branch with base:main.