# Hugging Face — 🧨 Accelerating Stable Diffusion XL Inference with JAX on Cloud TPU v5e

- Company: Hugging Face (huggingface.co)
- Announced: 2023-10-03
- Category: capability-change
- Coverage: not counted
- Announcement: yes
- Group: announcements
- Source: https://huggingface.co/blog/sdxl_jax
- Record: https://forck.live/items/1985-accelerating-stable-diffusion-xl-inference-with-jax-on-cloud-tpu-v5e
- Subject: Platform
- Models affected: Stable Diffusion XL, SDXL, stable-diffusion-xl-base-1.0

Hugging Face Diffusers now supports serving Stable Diffusion XL (SDXL) using JAX on Cloud TPU v5e, enabling high-performance, cost-efficient inference. The blog explains the benefits of JIT compilation and pmap parallelism for SDXL inference and provides code examples.

## Evidence

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

> Today, we are thrilled to announce that Hugging Face Diffusers now supports serving SDXL using JAX on Cloud TPUs, enabling high-performance, cost-efficient inference.

---

Record: https://forck.live/items/1985-accelerating-stable-diffusion-xl-inference-with-jax-on-cloud-tpu-v5e
Catalogue: https://forck.live/llms.txt
Current issue: https://forck.live/feed.md