An astronomer's introduction to NumPyro