-
Notifications
You must be signed in to change notification settings - Fork 100
Pull requests: google/orbax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Batch array persistence writes in CloudPathwaysArrayHandler.serialize.
#3580
opened Sep 17, 2026 by
copybara-service
Bot
Loading…
Add FileOptions.skip_sync_file_validations for async save
#3578
opened Sep 16, 2026 by
copybara-service
Bot
Loading…
Add pad_variable_length_inputs to BatchOptins and export configs
#3571
opened Sep 12, 2026 by
copybara-service
Bot
Loading…
Add offline converter for MTC checkpoints to standard Orbax format and end-to-end verification
#3570
opened Sep 11, 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…
Previous Next
ProTip!
Mix and match filters to narrow down what you’re looking for.