Skip to content

Pull requests: google/orbax

New pull request New
410 Open 2,878 Closed
410 Open 2,878 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

Read replicated safetensors tensors once and broadcast behind an option pull ready Ready to be pulled from GitHub into Google
#3475 opened Jul 14, 2026 by mridul-sahu Collaborator Loading…
5 of 8 tasks
Pick the safetensors over-read ratio per file and warn on heavy over-read pull ready Ready to be pulled from GitHub into Google
#3474 opened Jul 14, 2026 by mridul-sahu Collaborator Loading…
5 of 8 tasks
No public description
#3472 opened Jul 13, 2026 by copybara-service Bot Loading…
No public description
#3471 opened Jul 13, 2026 by copybara-service Bot Loading…
Internal
#3470 opened Jul 13, 2026 by copybara-service Bot Loading…
1
No public description
#3469 opened Jul 13, 2026 by copybara-service Bot Loading…
1
Internal
#3468 opened Jul 13, 2026 by copybara-service Bot Loading…
Internal
#3467 opened Jul 13, 2026 by copybara-service Bot Loading…
Internal
#3466 opened Jul 13, 2026 by copybara-service Bot Loading…
Internal
#3465 opened Jul 13, 2026 by copybara-service Bot Loading…
Internal
#3464 opened Jul 13, 2026 by copybara-service Bot Loading…
Add docstrings and type hints to multihost synchronization functions.
#3463 opened Jul 13, 2026 by copybara-service Bot Loading…
1
Add Colab, Kaggle, and GitHub badges to Orbax v1 checkpoint guides.
#3462 opened Jul 13, 2026 by copybara-service Bot Loading…
1
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…
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…
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…
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
ProTip! Updated in the last three days: updated:>2026-07-11.