Skip to content

Pull requests: google/orbax

New pull requestNew
403 Open 2,937 Closed
403 Open 2,937 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 reviewsReview requiredApproved reviewChanges requested
Assignee
Filter by who’s assigned
Assigned to nobodyLoading
Sort
Sort by
NewestOldestMost commentedLeast commentedRecently updatedLeast recently updatedBest match
Most reactions
👍👎😄🎉😕❤️🚀👀

Pull requests list

Workflow update.
#3534 opened Aug 12, 2026 by copybara-serviceBotLoading…
Delete unused orbax-level changelog (in favor of orbax/checkpoint-level changelog).
#3533 opened Aug 11, 2026 by copybara-serviceBotLoading…
No public description
#3531 opened Aug 11, 2026 by copybara-serviceBotLoading…
No public description
#3526 opened Aug 10, 2026 by copybara-serviceBotLoading…
No public description
#3523 opened Aug 7, 2026 by copybara-serviceBotLoading…
1
Split shardable arrays by mesh axis during snapshot saving.
#3516 opened Aug 4, 2026 by copybara-serviceBotLoading…
Cache sharding metadata device mesh serialization pull ready Ready to be pulled from GitHub into Google
#3514 opened Aug 2, 2026 by Trighap52Loading…
1
Update neptune model save api to take multiple TensorflowModel/JaxModel/CompositeModel as input.
#3512 opened Jul 31, 2026 by copybara-serviceBotLoading…
Add a migration guide for Neptune model export APIs to README.md.
#3511 opened Jul 31, 2026 by copybara-serviceBotLoading…
No public description
#3504 opened Jul 29, 2026 by copybara-serviceBotLoading…
No public description
#3501 opened Jul 29, 2026 by copybara-serviceBotLoading…
Enhance Orbax save stats granularity with fine-grained duration metrics.
#3498 opened Jul 28, 2026 by copybara-serviceBotLoading…
Handle None dimensions when converting to jax.ShapeDtypeStruct.
#3496 opened Jul 27, 2026 by copybara-serviceBotLoading…
Fix Orbax export compatibility with JAX ShapeDtypeStruct validation.
#3495 opened Jul 27, 2026 by copybara-serviceBotLoading…
1
docs: fix typo in CONTRIBUTING.md pytype table row
#3487 opened Jul 23, 2026 by amir-rezaeiLoading…
1
No public description
#3482 opened Jul 21, 2026 by copybara-serviceBotLoading…
No public description
#3471 opened Jul 13, 2026 by copybara-serviceBotLoading…
Internal
#3470 opened Jul 13, 2026 by copybara-serviceBotLoading…
1
No public description
#3469 opened Jul 13, 2026 by copybara-serviceBotLoading…
1
Internal
#3468 opened Jul 13, 2026 by copybara-serviceBotLoading…
Internal
#3467 opened Jul 13, 2026 by copybara-serviceBotLoading…
Internal
#3466 opened Jul 13, 2026 by copybara-serviceBotLoading…
Address blocker where benchmark runs failed to initialize JAX on Viperlite (vlp) and Ghostlite (glp) pods on Borg.
#3457 opened Jul 9, 2026 by copybara-serviceBotLoading…
Internal
#3449 opened Jul 7, 2026 by copybara-serviceBotLoading…
1
No public description
#3445 opened Jul 7, 2026 by copybara-serviceBotLoading…
1
ProTip! Type gi on any issue or pull request to go back to the issue listing page.