Modules:
| Name | Description |
|---|---|
abstract | |
cudnn_attn | |
flash_attn | |
flashinfer_attn | |
registry | Diffusion attention backend registry. |
ring | |
ring_flash_attn | |
ring_pytorch_attn | |
sage_attn | |
sage_attn3 | |
sdpa | |
utils | Utils for attention backends. |