Skip to content

Pull requests: google/orbax

New pull request New
344 Open 2,565 Closed
344 Open 2,565 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

Enable Memory Regulator for Orbax checkpoint saving.
#3078 opened Apr 11, 2026 by copybara-service bot Loading…
Use ParamInfo.keypath in v1 compatibility layer.
#3077 opened Apr 10, 2026 by copybara-service bot Loading…
Enable rich type support by default in PyTreeMetadataOptions.
#3076 opened Apr 10, 2026 by copybara-service bot Loading…
Remove load_all_checkpoint_weights option.
#3075 opened Apr 10, 2026 by copybara-service bot Loading…
Internal change.
#3074 opened Apr 9, 2026 by copybara-service bot Loading…
Add an option to clean the directory after each benchmark run.
#3073 opened Apr 9, 2026 by copybara-service bot Loading…
Add generic options field to SerializationParam to pass SaveArgs through.
#3072 opened Apr 9, 2026 by copybara-service bot Loading…
Enable Memory Regulator for Orbax checkpoint saving.
#3069 opened Apr 8, 2026 by copybara-service bot Loading…
Add AUTO resolution mode for pytree loading.
#3068 opened Apr 8, 2026 by copybara-service bot Loading…
Add multi-host support to Safetensors loading in Orbax.
#3060 opened Apr 6, 2026 by copybara-service bot Loading…
No public description
#3058 opened Apr 6, 2026 by copybara-service bot Loading…
No public description
#3057 opened Apr 6, 2026 by copybara-service bot Loading…
Simplify JAX distributed initialization in benchmarks.
#3056 opened Apr 6, 2026 by copybara-service bot Loading…
Enable rich type support by default in PyTreeMetadataOptions.
#3050 opened Apr 1, 2026 by copybara-service bot Loading…
Add monitoring for foreground operations in async checkpointing.
#3049 opened Apr 1, 2026 by copybara-service bot Loading…
Pin xpk version in Orbax testing Dockerfile.
#3047 opened Mar 31, 2026 by copybara-service bot Loading…
Optimize safetensors loading with 1D contiguous reads and ICI resharding.
#3046 opened Mar 31, 2026 by copybara-service bot Loading…
Consolidate FilesystemType resolution.
#3045 opened Mar 30, 2026 by copybara-service bot Loading…
Make CPU devices report as cpu:0 instead of TFRT_CPU_0.
#3041 opened Mar 27, 2026 by copybara-service bot Loading…
Centralize array storage options and implement field-level merging in Orbax v1.
#3039 opened Mar 26, 2026 by copybara-service bot Loading…
No public description
#3038 opened Mar 26, 2026 by copybara-service bot Loading…
No public description
#3037 opened Mar 26, 2026 by copybara-service bot Loading…
No public description
#3036 opened Mar 26, 2026 by copybara-service bot Loading…
No public description
#3035 opened Mar 25, 2026 by copybara-service bot Loading…
No public description
#3033 opened Mar 25, 2026 by copybara-service bot Loading…
ProTip! Updated in the last three days: updated:>2026-04-09.