flax
c2ba4a38
- 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: 332141521
References
#466 - Use jax.api.device_put_sharded() in place of private JAX APIs.
Author
Jake VanderPlas
Committer
a-googler
Parents
604112f9
Loading