- Notifications
You must be signed in to change notification settings - Fork44
Pull requests: google/orbax
Author
Label
Projects
Milestones
Reviews
Assignee
Assigned to nobodyLoading
Sort
Pull requests list
Broadcast interval time-based checkpoint decision across hosts in CheckpointManager from primary host.
#1742 openedMar 25, 2025 bycopybara-servicebotLoading…
Broadcast interval time-based checkpoint decision across hosts in CheckpointManager from primary host.
#1741 openedMar 25, 2025 bycopybara-servicebotLoading…
Patch for testing b/404645078, hardcoding single_host_load_and_broadcast=true.
#1740 openedMar 25, 2025 bycopybara-servicebotLoading…
Remove Module class and replace all the places using the Module to the dictionary.
#1739 openedMar 25, 2025 bycopybara-servicebotLoading…
Use ocp.futures.CommitFutureAwaitingContractedSignals to support async directory creation feature for pathways type handlers.
#1738 openedMar 24, 2025 bycopybara-servicebotLoading…
Use
ocp.future.CommitFutureAwaitingContractedSignals in custom async checkpoint handlers to support async directory creation. #1737 openedMar 24, 2025 bycopybara-servicebotLoading…
Wrap synchronous checkpoint handler calls in composite checkpoint handler with
CommitFutureAwaitingContractedSignals to not break save. #1736 openedMar 24, 2025 bycopybara-servicebotLoading…
Add
protobuf to Orbax Export dependencies and install protobuf using python and ensureprotoc installation in github workflow. #1729 openedMar 20, 2025 bycopybara-servicebotLoading…
Improve
NameFormat.find_all performance usingsingle_host_load_and_broadcast strategy. #1722 openedMar 19, 2025 bycopybara-servicebotLoading…
Fix the args ordering when using assertProto2Equal for OBM related tests
#1700 openedMar 12, 2025 bycopybara-servicebotLoading…
Allow configuring the minimum interval seconds for continuous checkpointing.
#1666 openedMar 6, 2025 bycopybara-servicebotLoading…
Simplify heartbeat options for PjRt distributed runtime.
#1663 openedMar 5, 2025 bycopybara-servicebotLoading…
Support minimum_interval_seconds since last save in Continuous checkpoint save policy.
#1645 openedMar 3, 2025 bycopybara-servicebotLoading…
ProTip! Find all pull requests that aren't related to any open issues with-linked:issue.