Stable Diffusion in JAX / Flax: Generate 8 Images in 8 Seconds with Google Colab and TPU Runtime