jax
b2d20891
- [JAX] Fail gracefully when an array with multiple shards is passed to make_array_from_single_device_arrays.
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
[JAX] Fail gracefully when an array with multiple shards is passed to make_array_from_single_device_arrays. Fix a crash when an exception is thrown during PyArray construction. PiperOrigin-RevId: 639760114
Author
hawkinsp
Committer
a-googler
Parents
fe35acd4
Loading