Skip to content

Pull requests: google/orbax

New pull request New
359 Open 2,654 Closed
359 Open 2,654 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

Remove unnecessary checkpointer testing files/subclassing in favor of new compatibility testing.
#3187 opened May 12, 2026 by copybara-service Bot Loading…
Add checkpointer compatibilty testing + remove redundant deps.
#3186 opened May 11, 2026 by copybara-service Bot Loading…
Fix _single_replica_deserialize_and_broadcast when active mesh is set.
#3185 opened May 11, 2026 by copybara-service Bot Loading…
No public description
#3180 opened May 8, 2026 by copybara-service Bot Loading…
Update async save response to add initiated property + fix returned .result() bool.
#3176 opened May 7, 2026 by copybara-service Bot Loading…
No public description
#3174 opened May 7, 2026 by copybara-service Bot Loading…
Add compatibilty tests to test against generated checkpoints + remove unnecessary testing files/subclassing.
#3170 opened May 7, 2026 by copybara-service Bot Loading…
Generate static checkpoint managed checkpoints using generate_v0_checkpoint_manager_checkpoints and generate_v1_checkpointer_checkpoints scripts using the following commands:
#3169 opened May 7, 2026 by copybara-service Bot Loading…
Add v0 checkpoint manager and v1 checkpointer checkpoint generation scripts.
#3168 opened May 7, 2026 by copybara-service Bot Loading…
Prioritize StatefulCheckpointable handlers in registration.
#3161 opened May 5, 2026 by copybara-service Bot Loading…
[monitoring] Pass config-managed key-value tags to monitoring listeners.
#3153 opened May 4, 2026 by copybara-service Bot Loading…
Add a script for converting Safetensors to Orbax native layout.
#3147 opened Apr 30, 2026 by copybara-service Bot Loading…
Add optimized GCS loading for Safetensors using gcsfuse.
#3138 opened Apr 28, 2026 by copybara-service Bot Loading…
Adds configurations for running p2p and Pathways-based checkpoint manager benchmarks on multi-slice v5litepod-16 TPUs.
#3137 opened Apr 28, 2026 by copybara-service Bot Loading…
Internal change.
#3134 opened Apr 27, 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…
Support saving host-local jax.Arrays in multi-process environments.
#3123 opened Apr 23, 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…
4
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…
ProTip! Updated in the last three days: updated:>2026-05-09.