Accelerate deep learning and other number-intensive tasks with JAX, Google’s awesome high-performance numerical computing library. In Deep Learning with JAX …