Train neural networks faster with JAX on TPUs
March 2, 2022
By
Jasper Van den Bossche
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.
February 3, 2022