Skip to content

Pull requests: google/orbax

New pull request New
286 Open 2,315 Closed
286 Open 2,315 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

Refactor CheckpointLayout splitting load() into load_pytree() and load_checkpointables() each with their own dedicated loading logic.
#2764 opened Jan 9, 2026 by copybara-service bot Loading…
5/N DAG orchestration support.
#2763 opened Jan 9, 2026 by copybara-service bot Loading…
Extract JAX model input signature calculation.
#2762 opened Jan 9, 2026 by copybara-service bot Loading…
4/N DAG orchestration support.
#2761 opened Jan 9, 2026 by copybara-service bot Loading…
Improve multislice zero buffer creation during broadcasting.
#2758 opened Jan 8, 2026 by copybara-service bot Loading…
Refactor CheckpointLayout to be stateless with respect to the path.
#2757 opened Jan 8, 2026 by copybara-service bot Loading…
Refactor CheckpointLayout to be stateless with respect to the path.
#2756 opened Jan 8, 2026 by copybara-service bot Loading…
Default cluster name and improve command output in XPK launch script.
#2755 opened Jan 8, 2026 by copybara-service bot Loading…
1
Improve restoring from single replica
#2754 opened Jan 8, 2026 by findmyway Loading…
internal
#2753 opened Jan 7, 2026 by copybara-service bot Loading…
1
Fix absolute link references to use relative path links.
#2748 opened Jan 6, 2026 by copybara-service bot Loading…
Drop support for Python 3.10.
#2746 opened Jan 6, 2026 by copybara-service bot Loading…
Add pytree_checkpoint_handler_test.py to multiprocess tests.
#2745 opened Jan 6, 2026 by copybara-service bot Loading…
1
Add save method to CheckpointLayout interface and implementations.
#2743 opened Jan 5, 2026 by copybara-service bot Loading…
Fix a type incompatibility issues.
#2741 opened Dec 23, 2025 by copybara-service bot Loading…
Add network metrics monitor in benchmark library.
#2740 opened Dec 23, 2025 by copybara-service bot Loading…
Adding AWS S3 support
#2739 opened Dec 22, 2025 by haoliuhl Loading…
1
3/N DAG orchestration support.
#2735 opened Dec 19, 2025 by copybara-service bot Loading…
Add dtype to Fragment classes.
#2733 opened Dec 18, 2025 by copybara-service bot Loading…
Move Orbax checkpoint benchmark configs to a new google/correctness subdirectory.
#2731 opened Dec 18, 2025 by copybara-service bot Loading…
1
Add validation for restoring with target trees containing none leaf types
#2727 opened Dec 17, 2025 by copybara-service bot Loading…
Update libs/dependency visibility
#2719 opened Dec 16, 2025 by copybara-service bot Loading…
1
Internal change
#2718 opened Dec 16, 2025 by copybara-service bot Loading…
#v1 Add a fresh implementation of StringLeafHandler to reduce unnecessary delegation complexity. Also introduce a typestr function for LeafHandler.
#2713 opened Dec 16, 2025 by copybara-service bot Loading…
obm: Refactor tf2obm interface
#2710 opened Dec 16, 2025 by copybara-service bot Loading…
ProTip! Follow long discussions with comments:>50.