-
Notifications
You must be signed in to change notification settings - Fork 89
Pull requests: google/orbax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
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…
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…
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…
Add colocated runtime helpers for Pathways MTC.
#3027
opened Mar 25, 2026 by
copybara-service
bot
Loading…
Implement auto-scaling for
save_device_host_concurrent_bytes.
#3015
opened Mar 23, 2026 by
copybara-service
bot
Loading…
security: add class-loading allowlist to PyTorch unpicklers (CWE-502)
#3014
opened Mar 23, 2026 by
SnailSploit
Loading…
Previous Next
ProTip!
Add no:assignee to see everything that’s not assigned.