Skip to content

Pull requests: google/orbax

New pull request New
176 Open 1,598 Closed
176 Open 1,598 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

#v1 Add v0 to v1 migration guide for load_pytree and load_checkpointables.
#1909 opened May 7, 2025 by copybara-service bot Loading…
Propagate JaxRuntimeError to callers in SignalingClient.
#1908 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 PreservationPolicy to v1, use PolicyCheckpointInfo instead of CheckpointInfo.
#1905 opened May 6, 2025 by copybara-service bot Loading…
Fix Random Key restoration with StandardRestore
#1902 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 v1 ArrayLeafHandler
#1900 opened May 5, 2025 by copybara-service bot Loading…
#v1 Add start_async_mkdir that begins async directory creation and returns a PathAwaitingCreation.
#1899 opened May 5, 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…
#v1 Add guide for ocp.training package.
#1893 opened May 5, 2025 by copybara-service bot Loading…
Enforce that the input and output name are not optional and remove the logic to deal with non input name and output name logic.
#1889 opened May 1, 2025 by copybara-service bot Loading…
1
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…
Don't raise errors when directories don't exist.
#1858 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
ProTip! Exclude everything labeled bug with -label:bug.