Skip to content

Pull requests: google/orbax

New pull request New
186 Open 1,730 Closed
186 Open 1,730 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

This should be enough to expose all the public things needed for users.
#2056 opened Jun 29, 2025 by copybara-service bot Loading…
Add verbose logging to the saving and preservation policies.
#2055 opened Jun 29, 2025 by copybara-service bot Loading…
Update AbstractScalar to a Type
#2054 opened Jun 27, 2025 by copybara-service bot Loading…
#v1 Add dummy orbax.checkpoint file to help identify checkpoints.
#2051 opened Jun 26, 2025 by copybara-service bot Loading…
#v1 Add checkpointables customization guide.
#2050 opened Jun 26, 2025 by copybara-service bot Loading…
Support for preservation policies in RCM
#2049 opened Jun 26, 2025 by copybara-service bot Loading…
1
Internal change.
#2047 opened Jun 25, 2025 by copybara-service bot Loading…
[OBM] Adding device assignment by coords to OBM
#2045 opened Jun 25, 2025 by copybara-service bot Loading…
1
Use force=True when saving from CheckpointManager, since existence checks are already performed.
#2044 opened Jun 24, 2025 by copybara-service bot Loading…
Support shardy in orbax.
#2041 opened Jun 24, 2025 by copybara-service bot Loading…
Dataset_Iterator Support added for Replicated CM
#2033 opened Jun 19, 2025 by copybara-service bot Loading…
1
draft, do not merge, just testing
#2031 opened Jun 18, 2025 by samos123 • Draft
[#v1](https://critique.corp.google.com/search/tag:v1) Add guide for Checkpointing & Exporting JAX Models package.
#2027 opened Jun 17, 2025 by copybara-service bot Loading…
1
Dataset Iterator support in ECM
#2026 opened Jun 17, 2025 by copybara-service bot Loading…
1
#emergency Eliminate usage of slice_index device property.
#2025 opened Jun 17, 2025 by copybara-service bot Loading…
Bugfixes to local checkpoint manager
#2012 opened Jun 16, 2025 by findmyway Loading…
2
**Changes**
#2009 opened Jun 13, 2025 by copybara-service bot Loading…
1
Remove old heartbeat options.
#2007 opened Jun 12, 2025 by copybara-service bot Loading…
Use heartbeat_timeout argument.
#2005 opened Jun 12, 2025 by copybara-service bot Loading…
Divide values into chunks when communicating with JAX distributed client
#1972 opened May 31, 2025 by copybara-service bot Loading…
1
Internal
#1967 opened May 29, 2025 by copybara-service bot Loading…
1
internal monitoring and hashing change
#1966 opened May 28, 2025 by copybara-service bot Loading…
1
Internal change.
#1964 opened May 28, 2025 by copybara-service bot Loading…
1
Handle the special case of _get_concrete_function when f is already a PolymorphicFunction.
#1958 opened May 23, 2025 by copybara-service bot Loading…
1
add the servo signature to the Pipeline.
#1930 opened May 12, 2025 by copybara-service bot Loading…
1
ProTip! Updated in the last three days: updated:>2025-06-27.