-
Notifications
You must be signed in to change notification settings - Fork 96
Pull requests: google/orbax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
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
#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…
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…
#v1 Allow PyTreeCheckpointHandler to save/load bare leaves.
#3284
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…
#v1 Add stringent validation for state abstract and concrete leaf types.
#3273
opened May 29, 2026 by
copybara-service
Bot
Loading…
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…
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…
Previous Next
ProTip!
Updated in the last three days: updated:>2026-05-31.