-
Notifications
You must be signed in to change notification settings - Fork 101
Pull requests: google/orbax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Refactor Orbax save metrics to Blocking vs. Background thread hierarchy.
#3540
opened Aug 14, 2026 by
copybara-service
Bot
Loading…
Detect GCSFuse mounts and default to commit-file atomicity
pull ready
Ready to be pulled from GitHub into Google
#3537
opened Aug 13, 2026 by
mridul-sahu
Collaborator
Loading…
5 of 8 tasks
Split shardable arrays by mesh axis during snapshot saving.
#3516
opened Aug 4, 2026 by
copybara-service
Bot
Loading…
Cache sharding metadata device mesh serialization
pull ready
Ready to be pulled from GitHub into Google
#3514
opened Aug 2, 2026 by
Trighap52
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…
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!
What’s not been updated in a month: updated:<2026-07-16.