Skip to content

Pull requests: google/orbax

New pull request New
392 Open 2,823 Closed
392 Open 2,823 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

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
3
Refactor per-worker save/load GB and GB/s metrics. These are now framed as global statistics, logged from a leader process, in contrast to /jax/orbax/*/worker/* metrics. In particular, delegation to TypeHandlers for memory size is eliminated and we rely instead on more expedient logic to determine global array sizes. This approach also has the virtue of being Pathways / mcJAX independent.
#3399 opened Jun 22, 2026 by copybara-service Bot Loading…
Test
#3398 opened Jun 22, 2026 by copybara-service Bot Loading…
Refactor Layouts to better distinguish between flat and composite checkpoints.
#3397 opened Jun 22, 2026 by copybara-service Bot Loading…
HuggingFace safetensors load benchmark pull ready Ready to be pulled from GitHub into Google
#3396 opened Jun 22, 2026 by mridul-sahu Collaborator Loading…
4 of 8 tasks
Internal.
#3394 opened Jun 21, 2026 by copybara-service Bot Loading…
Pin JAX dependency to version 0.10.1.
#3393 opened Jun 19, 2026 by copybara-service Bot Loading…
Use 127.0.0.1 instead of localhost for multihost testing.
#3391 opened Jun 18, 2026 by copybara-service Bot Loading…
Fix TPU bounds
#3389 opened Jun 18, 2026 by copybara-service Bot Loading…
Internal change.
#3388 opened Jun 17, 2026 by copybara-service Bot Loading…
Internal.
#3384 opened Jun 17, 2026 by copybara-service Bot Loading…
#cleanup Shut down BackgroundThreadRunner if result raises an exception.
#3381 opened Jun 16, 2026 by copybara-service Bot Loading…
Implement host-pinned transfer microbenchmarks and PyTree checkpoint integrations for Orbax benchmarks.
#3380 opened Jun 16, 2026 by copybara-service Bot Loading…
Internal
#3379 opened Jun 16, 2026 by copybara-service Bot Loading…
1
Reduce misleading warning messages in docs.
#3377 opened Jun 16, 2026 by copybara-service Bot Loading…
Roll forward of shape reporting with a fix for UnparsedFlagAccessError
#3374 opened Jun 16, 2026 by copybara-service Bot Loading…
1
Internal
#3373 opened Jun 16, 2026 by copybara-service Bot Loading…
Internal
#3372 opened Jun 16, 2026 by copybara-service Bot Loading…
Reduce misleading warning messages in docs.
#3371 opened Jun 16, 2026 by copybara-service Bot Loading…
Internal
#3367 opened Jun 15, 2026 by copybara-service Bot Loading…
1
Remove emergency directory from Orbax copybara exclusions.
#3365 opened Jun 15, 2026 by copybara-service Bot Loading…
Internal change.
#3363 opened Jun 12, 2026 by copybara-service Bot Loading…
Internal
#3361 opened Jun 12, 2026 by copybara-service Bot Loading…
Improve cancellation handling in Orbax Checkpoint V1.
#3358 opened Jun 12, 2026 by copybara-service Bot Loading…
ProTip! Type g i on any issue or pull request to go back to the issue listing page.