Jasper Van den Bossche

Train neural networks faster with JAX on TPUs

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.