Skip to content

Pull requests: google/orbax

New pull request New
145 Open 1,371 Closed
145 Open 1,371 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
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Add ArrayStorageOptions to customize per-leaf saving behavior for arrays (e.g. dtype).
#1642 opened Feb 28, 2025 by copybara-service bot Loading…
Internal change.
#1641 opened Feb 27, 2025 by copybara-service bot Loading…
Introduce retry module to retry coroutines.
#1639 opened Feb 27, 2025 by copybara-service bot Loading…
Enable ArrayMetadataStore by default
#1636 opened Feb 26, 2025 by copybara-service bot Loading…
Internal.
#1634 opened Feb 26, 2025 by copybara-service bot Loading…
Introduce Context and its usage as a contextmanager.
#1632 opened Feb 26, 2025 by copybara-service bot Loading…
internal
#1631 opened Feb 25, 2025 by copybara-service bot Loading…
Make the create_directories_asynchronously option default to True if jax distributed client is initialized.
#1630 opened Feb 25, 2025 by copybara-service bot Loading…
Add save barrier to ensure checkpoint manager and checkpointers save start is synchronized on all processes.
#1629 opened Feb 25, 2025 by copybara-service bot Loading…
Add CheckpointableHandler Protocol.
#1626 opened Feb 21, 2025 by copybara-service bot Loading…
Internal change.
#1625 opened Feb 21, 2025 by copybara-service bot Loading…
internal
#1620 opened Feb 19, 2025 by copybara-service bot Loading…
internal
#1615 opened Feb 18, 2025 by copybara-service bot Loading…
Add support to create dummy directories in orbax benchmark.
#1613 opened Feb 18, 2025 by copybara-service bot Loading…
Internal.
#1598 opened Feb 10, 2025 by copybara-service bot Loading…
Support single-slice checkpointing in emergency.CheckpointManager. Associated changes to adjust docstrings and AbstractCheckpointManager signatures.
#1593 opened Feb 7, 2025 by copybara-service bot Loading…
Some internal changes.
#1579 opened Feb 5, 2025 by copybara-service bot Loading…
Support single-slice checkpointing in emergency.CheckpointManager. Associated changes to adjust docstrings and AbstractCheckpointManager signatures.
#1574 opened Feb 4, 2025 by copybara-service bot Loading…
Nothing here, but needed for diffbase
#1567 opened Jan 30, 2025 by copybara-service bot Loading…
Add benchmarks comparing with/without asynchronous directory creation
#1561 opened Jan 30, 2025 by copybara-service bot Loading…
Add pytree_checkpoint_handler_test.
#1556 opened Jan 29, 2025 by copybara-service bot Loading…
Add checkpoint/proto/testing/BUILD.
#1545 opened Jan 27, 2025 by copybara-service bot Loading…
Internal
#1537 opened Jan 24, 2025 by copybara-service bot Loading…
1
Allow users to enable continuous checkpointing in Gemax Prod. Note that this is an experimental feature and should not be adopted by users without explicit approval.
#1530 opened Jan 23, 2025 by copybara-service bot Loading…
Add support for merging array groups across multiple devices, with uneven sharding between devices.
#1514 opened Jan 17, 2025 by copybara-service bot Loading…
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.