From the source
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.





