Skip to content

Pull requests: google/orbax

New pull request New
216 Open 1,287 Closed
216 Open 1,287 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 Github Action version to v6 and remove arduino/setup-protoc.
#3136 opened Apr 27, 2026 by copybara-service Bot Loading…
Internal Changes
#3135 opened Apr 27, 2026 by copybara-service Bot Loading…
Internal change.
#3134 opened Apr 27, 2026 by copybara-service Bot Loading…
fix github PR builds
#3132 opened Apr 27, 2026 by copybara-service Bot Loading…
Introduce repeating transformation operation.
#3131 opened Apr 27, 2026 by copybara-service Bot Loading…
No public description
#3130 opened Apr 27, 2026 by copybara-service Bot Loading…
Correctly use epath.Path.open as a context manager in async_open.
#3129 opened Apr 27, 2026 by copybara-service Bot Loading…
Implement Prometheus metrics emission for Orbax.
#3128 opened Apr 25, 2026 by copybara-service Bot Loading…
Internal Changes
#3127 opened Apr 24, 2026 by copybara-service Bot Loading…
Include p2p emergency checkpoint tests in OSS.
#3126 opened Apr 24, 2026 by copybara-service Bot Loading…
Implement Prometheus metrics emission for Orbax via JAX monitoring listeners.
#3125 opened Apr 24, 2026 by copybara-service Bot Loading…
No public description
#3124 opened Apr 23, 2026 by copybara-service Bot Loading…
1
Support saving host-local jax.Arrays in multi-process environments.
#3123 opened Apr 23, 2026 by copybara-service Bot Loading…
#v1 Add LeafHandler as a CheckpointableHandler, so that ordinary PyTree leaves can also be saved as individual checkpointables.
#3121 opened Apr 22, 2026 by copybara-service Bot Loading…
Not meant to be checked in.
#3119 opened Apr 22, 2026 by copybara-service Bot Loading…
Internal change.
#3114 opened Apr 22, 2026 by copybara-service Bot Loading…
No public description
#3110 opened Apr 20, 2026 by copybara-service Bot Loading…
Fix path-traversal vulnerability in emergency P2P checkpoint service
#3105 opened Apr 17, 2026 by YuvalElbar6 Loading…
Reduce default GCS target size to 400MiB for OCDBT in GCS
#3104 opened Apr 17, 2026 by copybara-service Bot Loading…
Removal of deprecated field
#3102 opened Apr 16, 2026 by copybara-service Bot Loading…
1
Add Safetensors to Orbax converter and improve GCS loading.
#3101 opened Apr 16, 2026 by copybara-service Bot Loading…
#v1 Add LeafHandler as a CheckpointableHandler, so that ordinary PyTree leaves can also be saved as individual checkpointables.
#3100 opened Apr 16, 2026 by copybara-service Bot Loading…
No public description
#3099 opened Apr 16, 2026 by copybara-service Bot Loading…
#v1 Allow a context to be default-configured for all Checkpointer operations.
#3098 opened Apr 15, 2026 by copybara-service Bot Loading…
Make sure single JAX.random.key can be save by PyTreeCheckpointHandler
#3097 opened Apr 15, 2026 by copybara-service Bot Loading…
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.