March 2, 2022
Training your neural network ten times faster using Jax on a TPU
Introduction blogpost to JAX : how it works, how it is different from Tensorflow/Pytorch and why we think every machine learning engineer should add it to its arsenal of tools.