Skip to content

Pull requests: google/orbax

New pull request New
173 Open 1,615 Closed
173 Open 1,615 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 v1 model surgery doc.
#1922 opened May 8, 2025 by copybara-service bot Loading…
#v1 Use an implementation of load independent of v0.Checkpointer.
#1921 opened May 8, 2025 by copybara-service bot Loading…
Enable pinned_host_transfer for GPU backend only. Internal reflection of https://github.com/google/orbax/pull/1910.
#1918 opened May 8, 2025 by copybara-service bot Loading…
Generalize supplemental info in the StableHLO body. This is to provide the flexibility to store more than one type of supplemental info in the future.
#1915 opened May 8, 2025 by copybara-service bot Loading…
Re-enable pinned_host transfers for GPU
#1910 opened May 7, 2025 by gspschmid Loading…
#v1 Add v0 to v1 migration guide for load_pytree and load_checkpointables.
#1909 opened May 7, 2025 by copybara-service bot Loading…
Move tree_difference and tree_trim to a separate structure_utils file.
#1907 opened May 6, 2025 by copybara-service bot Loading…
Add support for restoring jax random keys correctly in Standard Checkpoint Handler.
#1901 opened May 6, 2025 by copybara-service bot Loading…
Add tree_trim function that matches a tree's structure to a template.
#1897 opened May 5, 2025 by copybara-service bot Loading…
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…
Add new float8e3m4 dtype for testing
#1873 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…
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…
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…
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…
ProTip! no:milestone will show everything without a milestone.