Skip to content

Pull requests: google/orbax

New pull request New
295 Open 2,328 Closed
295 Open 2,328 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

Name output signature leaves in jax2obm based on their tree path.
#2786 opened Jan 16, 2026 by copybara-service bot Loading…
#v1 Move location of finalize log, as logging to early is confusing.
#2785 opened Jan 16, 2026 by copybara-service bot Loading…
Internal change
#2784 opened Jan 15, 2026 by copybara-service bot Loading…
Internal change
#2783 opened Jan 15, 2026 by copybara-service bot Loading…
Add context tensors to Kernel ExecuteContext and FlatSignature to runner ops.
#2782 opened Jan 15, 2026 by copybara-service bot Loading…
Internal testing.
#2781 opened Jan 15, 2026 by copybara-service bot Loading…
Internal change
#2780 opened Jan 14, 2026 by copybara-service bot Loading…
Get keyword_argument_execution for GraphBundle from pipeline proto.
#2779 opened Jan 14, 2026 by copybara-service bot Loading…
directory compatibilty in safetensorlayout
#2775 opened Jan 14, 2026 by copybara-service bot Loading…
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…
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
ProTip! Adding no:label will show everything without a label.