band_attention. A simple implementation of band attention with CUDA acceleration for faster Diffusion Transformers in sequential generation task.

github.com/luoshuqing2001/band_attention

Vaya's read on this project

Problem, audience, market, and the verdict — sign in to see it.

Updates

No recent activity.