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, how we are supporting JAX development, and give an overview of the ecosystem we are building to support researchers everywhere.Read More