JAX For Beginners, Novices and Experts

This is a short presentation with code snippets to introduce the JAX library

Requirements

jax : 0.4.23 numpy : 1.26.4 keras : 3.4.1 matplotlib: 3.8.4

Jupyter Notebook extensions

RISE - to render notebooks as reveal.js presentations
hide-code - to hide input commands