Skip to content

Pull requests: google/orbax

New pull request New
401 Open 2,893 Closed
401 Open 2,893 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

Handle PRNG keys in Snapshotter.
#3481 opened Jul 20, 2026 by copybara-service Bot Loading…
Make Snapshotter worker thread resilient to errors during jax.block_until_ready.
#3480 opened Jul 20, 2026 by copybara-service Bot Loading…
Internal
#3479 opened Jul 20, 2026 by copybara-service Bot Loading…
1
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
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…
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…
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
#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
#3435 opened Jul 3, 2026 by copybara-service Bot Loading…
1
No public description
#3433 opened Jul 3, 2026 by copybara-service Bot Loading…
1
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
Internal.
#3394 opened Jun 21, 2026 by copybara-service Bot Loading…
ProTip! Add no:assignee to see everything that’s not assigned.