Skip to content

Pull requests: google/orbax

New pull request New
165 Open 1,559 Closed
165 Open 1,559 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

Replace GlobalSupplemental & GlobalSupplementalMap with map<string, UnstructuredData> in manifest.
#1857 opened Apr 23, 2025 by copybara-service bot Loading…
#v1 Add documentation on checkpointables. This deals primarily with the concept of what a checkpointable is, as well as usage examples.
#1856 opened Apr 23, 2025 by copybara-service bot Loading…
#v1 Add checkpointables support for training.Checkpointer.
#1854 opened Apr 23, 2025 by copybara-service bot Loading…
Updated save and restore signatures for Emergency Checkpoint Manager
#1852 opened Apr 23, 2025 by copybara-service bot Loading…
1
Rename best_fn to get_metric_fn in CheckpointManagerOptions.
#1851 opened Apr 23, 2025 by copybara-service bot Loading…
#v1 Add compatibility tests for save-by-v0-load-by-v1 and also fix code.
#1849 opened Apr 22, 2025 by copybara-service bot Loading…
Internal only hashing and signing feature
#1842 opened Apr 18, 2025 by copybara-service bot Loading…
1
Remove hardcoded 0x777 file mode
#1834 opened Apr 15, 2025 by copybara-service bot Loading…
1
Internal change.
#1795 opened Apr 8, 2025 by copybara-service bot Loading…
Support lazy init of a CheckpointInfo with CheckpointManagerOptions.lazy_checkpoint_info.
#1794 opened Apr 8, 2025 by copybara-service bot Loading…
Log memory_info in save code path.
#1789 opened Apr 5, 2025 by copybara-service bot Loading…
Use the new preservation policy in CheckpointManager.
#1775 opened Apr 2, 2025 by copybara-service bot Loading…
Add tree_trim function for filtering a PyTree by another's structure.
#1764 opened Mar 28, 2025 by copybara-service bot Loading…
internal
#1761 opened Mar 28, 2025 by copybara-service bot Loading…
1
Catching very infrequent error in train-logs writing which crashes train job.
#1760 opened Mar 28, 2025 by copybara-service bot Loading…
1
Patch for use_replica_parallel=False.
#1751 opened Mar 26, 2025 by copybara-service bot Loading…
Patch for testing b/404645078, hardcoding single_host_load_and_broadcast=true.
#1740 opened Mar 25, 2025 by copybara-service bot Loading…
Use ocp.futures.CommitFutureAwaitingContractedSignals to support async directory creation feature for Gemax Prod
#1738 opened Mar 24, 2025 by copybara-service bot Loading…
Use ocp.future.CommitFutureAwaitingContractedSignals in custom async checkpoint handlers and type handlers to support async directory creation.
#1737 opened Mar 24, 2025 by copybara-service bot Loading…
Add protobuf to Orbax Export dependencies and install protobuf using python and ensure protoc installation in github workflow.
#1729 opened Mar 20, 2025 by copybara-service bot Loading…
Internal change
#1720 opened Mar 19, 2025 by copybara-service bot Loading…
1
Internal.
#1714 opened Mar 17, 2025 by copybara-service bot Loading…
Fix the args ordering when using assertProto2Equal for OBM related tests
#1700 opened Mar 12, 2025 by copybara-service bot Loading…
Allow configuring the minimum interval seconds for continuous checkpointing.
#1666 opened Mar 6, 2025 by copybara-service bot Loading…
1
Simplify heartbeat options for PjRt distributed runtime.
#1663 opened Mar 5, 2025 by copybara-service bot Loading…
ProTip! Follow long discussions with comments:>50.