Skip to content

Pull requests: google/orbax

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

Update TPU configurations for multiprocess tests.
#3392 opened Jun 18, 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…
No public description
#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…
Open source emergency checkpointing tests and fix OSS JAX simulation compatibility.
#3359 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…
ProTip! Follow long discussions with comments:>50.