'Deep Learning with JAX'
Accelerate deep learning and other number-intensive tasks with JAX, Googles awesome high-performance numerical computing library. In Deep Learning with JAX you will learn how to: Use JAX for numerical calculations Build differentiable models with JAX prim