flax
94f322a4
- Use jax.api.device_put_sharded() in place of private JAX apis.
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
5 years ago
Use jax.api.device_put_sharded() in place of private JAX apis. PiperOrigin-RevId: 332047658
References
test_332047658
Author
vanderplas
Committer
a-googler
Parents
604112f9
Loading