Skip to content

Pull requests: google/orbax

New pull request New
403 Open 2,870 Closed
403 Open 2,870 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
#3460 opened Jul 10, 2026 by copybara-service Bot Loading…
Avoid stale Pathways MTC worker dummy handles
#3459 opened Jul 9, 2026 by copybara-service Bot Loading…
Fix reached_preemption crash when JAX preemption service is disabled
#3458 opened Jul 9, 2026 by marsunique Loading…
3 of 8 tasks
1
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-service Bot Loading…
Drive safetensors load by the target sharding with bounded concurrent ranged reads pull ready Ready to be pulled from GitHub into Google
#3455 opened Jul 9, 2026 by mridul-sahu Collaborator Loading…
6 of 8 tasks
No public description
#3454 opened Jul 9, 2026 by copybara-service Bot Loading…
1
No public description
#3452 opened Jul 8, 2026 by copybara-service Bot Loading…
Update Orbax checkpointing and exporting guide to refer save and load instead of save_pytree and load_pytree.
#3451 opened Jul 8, 2026 by copybara-service Bot Loading…
1
Internal
#3449 opened Jul 7, 2026 by copybara-service Bot Loading…
1
No public description
#3445 opened Jul 7, 2026 by copybara-service Bot Loading…
1
Internal
#3444 opened Jul 7, 2026 by copybara-service Bot Loading…
Add custom_metadata argument to save methods in emergency checkpointing.
#3442 opened Jul 7, 2026 by copybara-service Bot Loading…
Internal
#3441 opened Jul 6, 2026 by copybara-service Bot Loading…
No public description
#3438 opened Jul 4, 2026 by copybara-service Bot Loading…
1
No public description
#3436 opened Jul 3, 2026 by copybara-service Bot Loading…
1
No public description
#3435 opened Jul 3, 2026 by copybara-service Bot Loading…
1
No public description
#3434 opened Jul 3, 2026 by copybara-service Bot Loading…
1
No public description
#3433 opened Jul 3, 2026 by copybara-service Bot Loading…
1
Internal.
#3431 opened Jul 2, 2026 by copybara-service Bot Loading…
Update Orbax Checkpoint v1 API docs for handler registration.
#3420 opened Jul 1, 2026 by copybara-service Bot Loading…
1
fix(security): pin etils-actions/pypi-auto-publish to commit SHA in publish workflows
#3414 opened Jun 28, 2026 by XananasX7 Loading…
1
Internal changes (moving code around)
#3407 opened Jun 25, 2026 by copybara-service Bot Loading…
No public description
#3405 opened Jun 25, 2026 by copybara-service Bot Loading…
Internal change.
#3401 opened Jun 22, 2026 by copybara-service Bot Loading…
Test PR to diagnose runner issue pull ready Ready to be pulled from GitHub into Google
#3400 opened Jun 22, 2026 by quoctruong Collaborator • Draft
5 of 8 tasks
13
ProTip! Adding no:label will show everything without a label.