JAX in 100 Seconds