Skip to content

Pull requests: google/orbax

New pull request New
352 Open 2,577 Closed
352 Open 2,577 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

#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…
No public description
#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…
Add multiprocess_test helper for Orbax OSS.
#3091 opened Apr 15, 2026 by copybara-service bot Loading…
No public description
#3090 opened Apr 15, 2026 by copybara-service bot Loading…
No public description
#3089 opened Apr 14, 2026 by copybara-service bot Loading…
#v1 Allow a callback to be registered for AsyncResponse that runs on completion.
#3088 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…
Enable Memory Regulator for Orbax checkpoint saving.
#3078 opened Apr 11, 2026 by copybara-service bot Loading…
Enable rich type support by default in PyTreeMetadataOptions.
#3076 opened Apr 10, 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…
Snapshot feature to save/load pathways checkpoint in host memory
#3056 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…
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.