Skip to content

Pull requests: google/orbax

New pull request New
298 Open 2,359 Closed
298 Open 2,359 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

#p2p implement Local and Persistent storage managers
#2820 opened Jan 27, 2026 by copybara-service bot Loading…
1
Add V1 benchmark.
#2819 opened Jan 27, 2026 by copybara-service bot Loading…
#p2p add PeerSelector for topology-aware shard retrieval
#2818 opened Jan 26, 2026 by copybara-service bot Loading…
1
#p2p implement P2PNode sidecar service
#2816 opened Jan 26, 2026 by copybara-service bot Loading…
1
Enforce topology check in StandardCheckpointHandler.restore and provide proper sharding fallback
#2815 opened Jan 26, 2026 by copybara-service bot Loading…
#p2p implement PeerRegistry for tracking peers
#2812 opened Jan 24, 2026 by copybara-service bot Loading…
1
Internal change.
#2811 opened Jan 23, 2026 by copybara-service bot Loading…
No public description
#2810 opened Jan 23, 2026 by copybara-service bot Loading…
Add support for loading graphs from pipeline components.
#2809 opened Jan 23, 2026 by copybara-service bot Loading…
Upgrade GitHub Actions for Node 24 compatibility
#2808 opened Jan 23, 2026 by salmanmkc Loading…
internal
#2806 opened Jan 22, 2026 by copybara-service bot Loading…
1
Validate all keyword arguments against the signature in the interpreter.
#2800 opened Jan 21, 2026 by copybara-service bot Loading…
Implement keyword argument support for graph orchestration.
#2797 opened Jan 21, 2026 by copybara-service bot Loading…
Plumb flat input/output signatures to TfSavedModelFunctionRunnerOp.
#2796 opened Jan 21, 2026 by copybara-service bot Loading…
Initialize the flat input/output signature for the shlo kernel based on the named signatures.
#2794 opened Jan 20, 2026 by copybara-service bot Loading…
Extract named signatures from Orbax pipelines.
#2792 opened Jan 20, 2026 by copybara-service bot Loading…
Internal change
#2791 opened Jan 20, 2026 by copybara-service bot Loading…
Allow mixing TensorSpec and TensorSpecWithDefault in Orbax Export.
#2788 opened Jan 16, 2026 by copybara-service bot Loading…
Add a key word to value map 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…
Get keyword_argument_execution for GraphBundle from pipeline proto.
#2779 opened Jan 14, 2026 by copybara-service bot Loading…
internal
#2773 opened Jan 13, 2026 by copybara-service bot Loading…
1
Add a skeleton for executing the graph with keyword arguments.
#2770 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…
ProTip! Follow long discussions with comments:>50.