Tutorial: JAX 101