Skip to content

Pull requests: google/orbax

New pull request New
201 Open 1,758 Closed
201 Open 1,758 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
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Internal change
#2101 opened Jul 12, 2025 by copybara-service bot Loading…
1
Refactor Snapshot class to be a stateful Protocol.
#2100 opened Jul 12, 2025 by copybara-service bot Loading…
Change recursively_copy_files parameter types from str to PathLike.
#2099 opened Jul 11, 2025 by copybara-service bot Loading…
Use HNS-compatible listing in NameFormat.
#2098 opened Jul 11, 2025 by copybara-service bot Loading…
Attempt to restore from single replica instead of single slice
#2096 opened Jul 11, 2025 by findmyway • Draft
[OBM] Deletes redundant tests as they are moved back to OBM
#2095 opened Jul 11, 2025 by copybara-service bot Loading…
1
Add branching logic to loading.py and refactor the checkpoint handler logic.
#2094 opened Jul 10, 2025 by copybara-service bot Loading…
Swap order of expected value and actual value when using [assertProto2Equal](https://source.corp.google.com/piper///depot/google3/net/proto2/contrib/pyutil/compare.py;rcl=781699748;l=153) to be consistent with definition.
#2092 opened Jul 10, 2025 by copybara-service bot Loading…
1
Remove oex_orchestration dependencies from obm test.
#2091 opened Jul 10, 2025 by copybara-service bot Loading…
1
Remove unique count from tmp checkpoint directory suffixes.
#2089 opened Jul 10, 2025 by copybara-service bot Loading…
1
add orbax_layout.py using the CheckpointLayout protocol.
#2088 opened Jul 9, 2025 by copybara-service bot Loading…
Make the temp directory generation be unique across CheckpointManager when shared barrier_sync_key_prefix.
#2086 opened Jul 9, 2025 by copybara-service bot Loading…
1
Restore from single slice
#2084 opened Jul 9, 2025 by findmyway • Draft
9
Add CheckpointLayout protocol for various checkpoint formats to inherit from.
#2082 opened Jul 7, 2025 by copybara-service bot Loading…
Add CheckpointLayout protocol for various checkpoint formats to inherit from.
#2081 opened Jul 7, 2025 by copybara-service bot Loading…
Update all V1 Codes work with V1 (De)SerializationParams and Metadata. Avoid mixing with V0 ones.
#2079 opened Jul 7, 2025 by copybara-service bot Loading…
Simplify is saving in progress by only tracking save status at primary host.
#2077 opened Jul 7, 2025 by copybara-service bot Loading…
Remove multiprocessing_options from TemporaryPath.
#2076 opened Jul 4, 2025 by copybara-service bot Loading…
Make TemporaryPath.create not multiprocessing-aware.
#2075 opened Jul 4, 2025 by copybara-service bot Loading…
Pr 2057
#2074 opened Jul 2, 2025 by deepikarajani24 Loading…
Add DescriptorHandler test. Also fix is_abstract_handleable method.
#2073 opened Jul 2, 2025 by copybara-service bot Loading…
Add tests for ProtoHandler.
#2071 opened Jul 2, 2025 by copybara-service bot Loading…
Internal change.
#2069 opened Jul 2, 2025 by copybara-service bot Loading…
Internal change.
#2067 opened Jul 1, 2025 by copybara-service bot Loading…
1
[OBM-Juno] Capture jax mesh in Genmedia exports
#2066 opened Jul 1, 2025 by copybara-service bot Loading…
1
ProTip! Exclude everything labeled bug with -label:bug.