Skip to content

Pull requests: google/orbax

New pull request New
355 Open 2,583 Closed
355 Open 2,583 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

No public description
#3108 opened Apr 17, 2026 by copybara-service bot Loading…
Convert local_checkpoint_directory to epath.Path.
#3107 opened Apr 17, 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 target size to 200MiB for OCDBT in GCS
#3104 opened Apr 17, 2026 by copybara-service bot Loading…
Support partial saves via StatefulCheckpointable wrapper and remove path check.
#3103 opened Apr 16, 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…
No public description
#3096 opened Apr 15, 2026 by copybara-service bot Loading…
Internal
#3095 opened Apr 15, 2026 by copybara-service bot Loading…
Support saving host-local jax.Arrays in multi-process environments.
#3094 opened Apr 15, 2026 by copybara-service bot Loading…
Add Snapshotter benchmarks and integrate Pathways initialization.
#3092 opened Apr 15, 2026 by copybara-service bot Loading…
Fix PRNG key deserialization failure.
#3089 opened Apr 14, 2026 by copybara-service bot Loading…
Rollback Jax/NumpyRandomKeyCheckpointHandler deprecation.
#3087 opened Apr 14, 2026 by copybara-service bot Loading…
No public description
#3085 opened Apr 13, 2026 by copybara-service bot Loading…
No public description
#3082 opened Apr 13, 2026 by copybara-service bot Loading…
Migrate Orbax Export to use Neptune orchestration protos.
#3081 opened Apr 13, 2026 by copybara-service bot Loading…
Remove load_all_checkpoint_weights option.
#3075 opened Apr 10, 2026 by copybara-service bot Loading…
Internal change.
#3074 opened Apr 9, 2026 by copybara-service bot Loading…
Add generic options field to SerializationParam to pass SaveArgs through.
#3072 opened Apr 9, 2026 by copybara-service bot Loading…
Enable Memory Regulator for Orbax checkpoint saving.
#3069 opened Apr 8, 2026 by copybara-service bot Loading…
No public description
#3058 opened Apr 6, 2026 by copybara-service bot Loading…
ProTip! Type g i on any issue or pull request to go back to the issue listing page.