Skip to content

Pull requests: google/orbax

New pull request New
341 Open 2,553 Closed
341 Open 2,553 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

Allow build_image.sh to use a local repo
#3063 opened Apr 7, 2026 by copybara-service bot Loading…
#v1 Add a path property to CheckpointMetadata. Separate step.NameFormat definitions from v0.
#3062 opened Apr 6, 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…
No public description
#3027 opened Mar 25, 2026 by copybara-service bot Loading…
Internal change.
#3023 opened Mar 24, 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…
1
security: remove allow_pickle=True from NumpyLayout (CVE-2019-6446 bypass)
#3013 opened Mar 23, 2026 by SnailSploit Loading…
1
No public description
#3010 opened Mar 20, 2026 by copybara-service bot Loading…
No public description
#3009 opened Mar 20, 2026 by copybara-service bot Loading…
1
ProTip! Mix and match filters to narrow down what you’re looking for.