Skip to content

Pull requests: google/orbax

New pull request New
298 Open 2,372 Closed
298 Open 2,372 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 only change.
#2833 opened Jan 29, 2026 by copybara-service bot Loading…
Update benchmark utility to support replicated arrays
#2830 opened Jan 28, 2026 by copybara-service bot Loading…
#v1 Clarify blocking behavior if a save is already ongoing for Checkpointer.
#2828 opened Jan 28, 2026 by copybara-service bot Loading…
Add Safetensors load benchmark.
#2827 opened Jan 28, 2026 by copybara-service bot Loading…
Update benchmark config and add pod killing script.
#2826 opened Jan 28, 2026 by copybara-service bot Loading…
1
#p2p implement Composite CheckpointManager
#2824 opened Jan 27, 2026 by copybara-service bot Loading…
1
Update Copybara to rewrite xla_data.proto import path.
#2823 opened Jan 27, 2026 by copybara-service bot Loading…
#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…
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
ProTip! Add no:assignee to see everything that’s not assigned.