co-author of JAX (https://github.com/jax-ml/jax)
https://cs.stanford.edu/~rfrostig
If you want to learn more, see this awesome book "How to Scale Your Model":
jax-ml.github.io/scaling-book/
Put together by several of my Google DeepMind colleagues listed below 🎉.
If you want to learn more, see this awesome book "How to Scale Your Model":
jax-ml.github.io/scaling-book/
Put together by several of my Google DeepMind colleagues listed below 🎉.
We hope that it helps you make the most of your computing resources. Enjoy!
We hope that it helps you make the most of your computing resources. Enjoy!