Using JAX to accelerate our research
Here we share our experience of working with JAX, outline why we find it useful for our AI research, and give an overview of the ecosystem we are building to support researchers everywhere.
Go to Source
https://deepmind.com/blog/article/using-jax-to-accelerate-our-research
