Skip to content

Pull requests: google/orbax

New pull request New
304 Open 2,383 Closed
304 Open 2,383 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 unused utility functions from Orbax Export.
#2851 opened Feb 4, 2026 by copybara-service bot Loading…
Fix argument handling in TfDataProcessor for single-argument tf.function.
#2850 opened Feb 4, 2026 by copybara-service bot Loading…
my public
#2849 opened Feb 4, 2026 by copybara-service bot Loading…
Just Update the README.md, CONTRIBUTING.md and CHANGELOG.md
#2848 opened Feb 4, 2026 by lshariprasad Loading…
3
Make RAMFS setup optional in TPU VM benchmarks.
#2847 opened Feb 4, 2026 by copybara-service bot Loading…
1
[Orbax] Optimize safetensors on-device loading by using a host RAM buffer.
#2846 opened Feb 4, 2026 by copybara-service bot Loading…
Add MANIFEST.in to include proto and pb2 files in Orbax Export.
#2844 opened Feb 3, 2026 by copybara-service bot Loading…
Testing unit tests
#2843 opened Feb 3, 2026 by copybara-service bot Loading…
1
Migrate python/ directory.
#2842 opened Feb 3, 2026 by copybara-service bot Loading…
1
#p2p Add Grain data iterator checkpointing to P2P
#2836 opened Feb 1, 2026 by copybara-service bot Loading…
1
my public
#2835 opened Jan 30, 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…
ECM Testing added in XPK
#2826 opened Jan 28, 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…
Add a V1 benchmark in the new benchmarking setup.
#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…
ProTip! Follow long discussions with comments:>50.