A pedagogical introduction to JAX/Equinox