Skip to content

Pull requests: google/orbax

New pull request New
389 Open 2,822 Closed
389 Open 2,822 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

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
No public description
#3378 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…
No public description
#3357 opened Jun 11, 2026 by copybara-service Bot Loading…
Internal change
#3356 opened Jun 11, 2026 by copybara-service Bot Loading…
use public path for tree_util.flatten_one_level
#3354 opened Jun 11, 2026 by copybara-service Bot Loading…
ProTip! Filter pull requests by the default branch with base:main.