-
Notifications
You must be signed in to change notification settings - Fork 103
Pull requests: google/orbax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Optimize dtype conversion in Orbax shard reads to resolve throughput bottleneck.
#3522
opened Aug 7, 2026 by
copybara-service
Bot
Loading…
[Neptune/Orbax] Add export-time per-output placement metadata to model manifest and ServingConfig
#3519
opened Aug 6, 2026 by
copybara-service
Bot
Loading…
Split shardable arrays by mesh axis during snapshot saving.
#3516
opened Aug 4, 2026 by
copybara-service
Bot
Loading…
Update neptune model save api to take multiple TensorflowModel/JaxModel/CompositeModel as input.
#3512
opened Jul 31, 2026 by
copybara-service
Bot
Loading…
Add a migration guide for Neptune model export APIs to README.md.
#3511
opened Jul 31, 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…
docs: fix typo in CONTRIBUTING.md pytype table row
#3487
opened Jul 23, 2026 by
amir-rezaei
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!
Add no:assignee to see everything that’s not assigned.