Skip to content

Pull requests: google/orbax

New pull request New
288 Open 2,323 Closed
288 Open 2,323 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

Remove deprecated obm_kwargs from ServingConfig.
#2774 opened Jan 14, 2026 by copybara-service bot Loading…
internal
#2773 opened Jan 13, 2026 by copybara-service bot Loading…
1
Add monitoring for Orbax checkpoint read performance on workers. This enables to collect per-worker metrics even on Pathways. In addition, we track the IO bytes requested from Tensorstore.
#2772 opened Jan 13, 2026 by copybara-service bot Loading…
Add a skeleton for executing the graph with keyword arguments.
#2770 opened Jan 13, 2026 by copybara-service bot Loading…
PUBLIC:6/N DAG orchestration support.
#2769 opened Jan 13, 2026 by copybara-service bot Loading…
Introduce the new proto 'Component' to represent the nodes in a pipeline.
#2767 opened Jan 12, 2026 by copybara-service bot Loading…
Fix Pytree Metadata resolution and refactor V0 Layout logic
#2766 opened Jan 12, 2026 by copybara-service bot Loading…
Add P2P CheckpointManager with persistent failover
#2765 opened Jan 12, 2026 by copybara-service bot Loading…
1
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…
Extract JAX model input signature calculation.
#2762 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.
#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…
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
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
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.