pymllm.layers.attention

Attention layers and backends for pymllm.

Submodules

• pymllm.layers.attention.attention_backend
• pymllm.layers.attention.flashinfer_backend
• pymllm.layers.attention.gdn
• pymllm.layers.attention.gdn_backend
• pymllm.layers.attention.gdn_chunkwise
• pymllm.layers.attention.hybrid_backend
• pymllm.layers.attention.radix_attention
• pymllm.layers.attention.radix_linear_attention