🧨 Stable Diffusion in JAX / Flax !

In three linesStable Diffusion is now available in JAX/Flax. This implementation provides improved performance and scalability for inference and training on TPU and GPU. Code is open-source on Hugging Face.

Summary generated by Claude — human-verified