Skip to content

Pull requests: google/orbax

New pull request New
180 Open 1,687 Closed
180 Open 1,687 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 old heartbeat options.
#2007 opened Jun 12, 2025 by copybara-service bot Loading…
Remove old heartbeat flags and arguments.
#2006 opened Jun 12, 2025 by copybara-service bot Loading…
Use heartbeat_timeout argument.
#2005 opened Jun 12, 2025 by copybara-service bot Loading…
Set heartbeat_timeout argument and flag.
#2004 opened Jun 12, 2025 by copybara-service bot Loading…
Avoid using the default key in obm_export.py.
#2002 opened Jun 12, 2025 by copybara-service bot Loading…
#v1 Modify saving code to eliminate cross-thread event loop sharing. Instead we create a background Thread and schedule tasks via run_coroutine_threadsafe. This returns a concurrent.futures.Future, which fits naturally with our AsyncResponse.
#2001 opened Jun 12, 2025 by copybara-service bot Loading…
Update logic for applying the default LatestN preservation policy in CM
#2000 opened Jun 12, 2025 by copybara-service bot Loading…
1
Enable keep_every_n GCS checkpoints in emergency checkpoint manager
#1998 opened Jun 11, 2025 by copybara-service bot Loading…
1
Fix Continuous Save Decision Policy.
#1993 opened Jun 10, 2025 by copybara-service bot Loading…
Add partial restore omission to v1 PyTreeHandler.
#1991 opened Jun 9, 2025 by copybara-service bot Loading…
Update pytree_handler to use the v0/v1 compatibility type handler registry.
#1988 opened Jun 9, 2025 by copybara-service bot Loading…
Expose keep_period when creating persistent checkpoint manager
#1987 opened Jun 9, 2025 by findmyway Loading…
Add Function::data_names to specify the data to load
#1982 opened Jun 5, 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
Use the new preservation policy in CheckpointManager.
#1880 opened Apr 29, 2025 by copybara-service bot Loading…
Re-organize OCP v0 and v1 docs directory structure.
#1876 opened Apr 28, 2025 by copybara-service bot Loading…
Removing _track_best from manager
#1870 opened Apr 27, 2025 by copybara-service bot Loading…
Add SaveDecisionPolicy and PreservationPolicy to v1, use CheckpointMetadata instead of CheckpointInfo.
#1862 opened Apr 24, 2025 by copybara-service bot Loading…
Rename best_fn to get_metric_fn in CheckpointManagerOptions.
#1851 opened Apr 23, 2025 by copybara-service bot Loading…
Remove hardcoded 0x777 file mode
#1834 opened Apr 15, 2025 by copybara-service bot Loading…
1
ProTip! Filter pull requests by the default branch with base:main.