-
Notifications
You must be signed in to change notification settings - Fork 102
Pull requests: google/orbax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Support storing OCDBT B-tree metadata and array data separately.
#3509
opened Jul 30, 2026 by
copybara-service
Bot
Loading…
Constrain jax data processors running on host only.
#3505
opened Jul 29, 2026 by
copybara-service
Bot
Loading…
Enhance Orbax save stats granularity with fine-grained duration metrics.
#3502
opened Jul 29, 2026 by
copybara-service
Bot
Loading…
Enhance Orbax save stats granularity with fine-grained duration metrics.
#3498
opened Jul 28, 2026 by
copybara-service
Bot
Loading…
Handle None dimensions when converting to jax.ShapeDtypeStruct.
#3496
opened Jul 27, 2026 by
copybara-service
Bot
Loading…
Fix Orbax export compatibility with JAX ShapeDtypeStruct validation.
#3495
opened Jul 27, 2026 by
copybara-service
Bot
Loading…
Add callback option to v1 MemoryOptions and propagate it to the array handler
#3491
opened Jul 27, 2026 by
copybara-service
Bot
Loading…
docs: fix typo in CONTRIBUTING.md pytype table row
#3487
opened Jul 23, 2026 by
amir-rezaei
Loading…
Bypass client-side readiness caching in Snapshotter using a donated JIT identity function.
#3484
opened Jul 23, 2026 by
copybara-service
Bot
Loading…
Read replicated safetensors tensors once and broadcast behind an option
pull ready
Ready to be pulled from GitHub into Google
#3475
opened Jul 14, 2026 by
mridul-sahu
Collaborator
Loading…
5 of 8 tasks
Fix reached_preemption crash when JAX preemption service is disabled
#3458
opened Jul 9, 2026 by
marsunique
Loading…
3 of 8 tasks
Address blocker where benchmark runs failed to initialize JAX on Viperlite (vlp) and Ghostlite (glp) pods on Borg.
#3457
opened Jul 9, 2026 by
copybara-service
Bot
Loading…
Previous Next
ProTip!
Exclude everything labeled
bug with -label:bug.