Introduction: JAX

Author

Marie-Hélène Burle

A brief introduction to JAX and to the JAX deep learning stack.

Slides (Click and wait: the presentation might take a few instants to load)