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

Hugging Face's Diffusers library now supports JAX/Flax for Stable Diffusion inference, enabling fast inference on Google TPUs.
From the source
🤗 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.
huggingface.co