Skip to content

Pull requests: google/orbax

New pull request New
377 Open 2,740 Closed
377 Open 2,740 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

Internal
#3302 opened Jun 2, 2026 by copybara-service Bot Loading…
Add per-host benchmark scalars, cross-host summaries, and checkpoint inventory cards pull ready Ready to be pulled from GitHub into Google
#3301 opened Jun 2, 2026 by mridul-sahu Collaborator Loading…
4 of 8 tasks
Modify the telemetry for saving metrics.
#3298 opened Jun 2, 2026 by copybara-service Bot Loading…
No public description
#3297 opened Jun 2, 2026 by copybara-service Bot Loading…
#v1 Remove test base classes for V1 Checkpoint API.
#3296 opened Jun 2, 2026 by copybara-service Bot Loading…
Reserving a new checkpoint type for an internal project.
#3295 opened Jun 2, 2026 by copybara-service Bot Loading…
1
Remove partial_save flag from metadata methods.
#3294 opened Jun 1, 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…
Adds orbax/checkpoint/experimental/v1/_src/handlers:"pytree_handler_partial_save_test to the presubmit tests.
#3279 opened Jun 1, 2026 by copybara-service Bot Loading…
Internal change.
#3275 opened May 29, 2026 by copybara-service Bot Loading…
#v1 Add stringent validation for state abstract and concrete leaf types.
#3273 opened May 29, 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
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…
ProTip! Updated in the last three days: updated:>2026-05-31.