Hugging Face Diffusers 支持在 JAX/Flax 上运行 Stable Diffusion

Hugging Face Diffusers v0.5.1 起支持 Flax，实现 Stable Diffusion 在 Google TPU 上的快速推理。该方案利用 bfloat16 精度和 8 个并行 TPU 设备，通过复制提示词使单批次生成 8 张图像，显著提升吞吐量。