Skip to content

Pull requests: google/orbax

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

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…
No public description
#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…
No public description
#3057 opened Apr 6, 2026 by copybara-service bot Loading…
Enable rich type support by default in PyTreeMetadataOptions.
#3050 opened Apr 1, 2026 by copybara-service bot Loading…
Add monitoring for foreground operations in async checkpointing.
#3049 opened Apr 1, 2026 by copybara-service bot Loading…
Pin xpk version in Orbax testing Dockerfile.
#3047 opened Mar 31, 2026 by copybara-service bot Loading…
ProTip! Adding no:label will show everything without a label.