Skip to content

Pull requests: google/orbax

New pull request New
387 Open 2,820 Closed
387 Open 2,820 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

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…
Add raw IO read / write per-worker metrics in jax_array_handlers.py.
#3369 opened Jun 15, 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…
Introduce Cancellation API for Orbax
#3353 opened Jun 11, 2026 by copybara-service Bot Loading…
Fork CloudPathwaysArrayHandler into Orbax from pathwaysutils.
#3351 opened Jun 11, 2026 by copybara-service Bot Loading…
ProTip! Adding no:label will show everything without a label.