在 TPU 上加速视频扩散模型的时空注意力

Accelerating Spatio-Temporal Attention for Video Diffusion on TPUs

Google researchers present a case study on optimizing video diffusion models by implementing sparse spatio-temporal attention on TPUs to achieve significant generation speedups.