modelsOct 13
🧨 Stable Diffusion in JAX / Flax !
Hugging Face has released a JAX/Flax implementation of Stable Diffusion. This allows for faster and more efficient deployment on TPUs and GPUs. The new implementation enables builders to leverage JAX's performance optimizations and Flax's ease of use.
Key takeaways
- Stable Diffusion now available in JAX/Flax.
- Enables faster deployment on TPUs and GPUs.
- Leverages JAX performance optimizations and Flax ease of use.