Skip to content

Pull requests: google/orbax

New pull request New
161 Open 1,455 Closed
161 Open 1,455 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

Adding preservation policy
#1747 opened Mar 26, 2025 by copybara-service bot Loading…
1
Remove Jax2obm_kwargs flag support_tf_resources now that it is supported by default and is always True.
#1746 opened Mar 25, 2025 by copybara-service bot Loading…
1
Automated g4 rollback of changelist 740040778.
#1743 opened Mar 25, 2025 by copybara-service bot Loading…
1
Broadcast interval time-based checkpoint decision across hosts in CheckpointManager from primary host.
#1742 opened Mar 25, 2025 by copybara-service bot Loading…
1
Patch for testing b/404645078, hardcoding single_host_load_and_broadcast=true.
#1740 opened Mar 25, 2025 by copybara-service bot Loading…
Remove Module class and replace all the places using the Module to the dictionary.
#1739 opened Mar 25, 2025 by copybara-service bot Loading…
1
Use ocp.futures.CommitFutureAwaitingContractedSignals to support async directory creation feature for pathways type handlers.
#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…
Wrap synchronous checkpoint handler calls in composite checkpoint handler with CommitFutureAwaitingContractedSignals to not break save.
#1736 opened Mar 24, 2025 by copybara-service bot Loading…
Introduce Placeholder to PyTreeCheckpointHandler. This allows PyTrees to be partially restored by specifying which nodes should be skipped using a Placeholder value.
#1732 opened Mar 20, 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
#1724 opened Mar 19, 2025 by copybara-service bot Loading…
1
Improve NameFormat.find_all performance using single_host_load_and_broadcast strategy.
#1722 opened Mar 19, 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…
Internal
#1707 opened Mar 13, 2025 by copybara-service bot Loading…
1
Fix the args ordering when using assertProto2Equal for OBM related tests
#1700 opened Mar 12, 2025 by copybara-service bot Loading…
#v1 Add metadata free functions.
#1691 opened Mar 10, 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…
Internal change.
#1662 opened Mar 5, 2025 by copybara-service bot Loading…
Internal
#1661 opened Mar 5, 2025 by copybara-service bot Loading…
1
Internal change.
#1649 opened Mar 3, 2025 by copybara-service bot Loading…
Internal change.
#1648 opened Mar 3, 2025 by copybara-service bot Loading…
internal
#1647 opened Mar 3, 2025 by copybara-service bot Loading…
ProTip! no:milestone will show everything without a milestone.