Accelerating Machine Learning research with JAX and TPU July 16, 2022Talk, Google I/O Extended Hanoi 2022, Hanoi