# Hugging Face — 🧨 Stable Diffusion in JAX / Flax !

- Company: Hugging Face (huggingface.co)
- Announced: 2022-10-13
- Category: capability-change
- Coverage: not counted
- Announcement: yes
- Group: announcements
- Source: https://huggingface.co/blog/stable_diffusion_jax
- Record: https://forck.live/items/2144-stable-diffusion-in-jax-flax
- Subject: Platform
- Models affected: CompVis/stable-diffusion-v1-4
- License: CreativeML OpenRAIL-M

Hugging Face's Diffusers library now supports JAX/Flax for Stable Diffusion inference, enabling fast inference on Google TPUs.

## Evidence

Verbatim from https://huggingface.co/blog/stable_diffusion_jax:

> 🤗 Hugging Face Diffusers supports Flax since version 0.5.1! This allows for super fast inference on Google TPUs, such as those available in Colab, Kaggle or Google Cloud Platform.

---

Record: https://forck.live/items/2144-stable-diffusion-in-jax-flax
Catalogue: https://forck.live/llms.txt
Current issue: https://forck.live/feed.md