All modules for which code is available
- xformers.components.attention
- xformers.components.attention.attention_mask
- xformers.components.attention.base
- xformers.components.attention.favor
- xformers.components.attention.global_tokens
- xformers.components.attention.linformer
- xformers.components.attention.local
- xformers.components.attention.nystrom
- xformers.components.attention.ortho
- xformers.components.attention.random
- xformers.components.attention.scaled_dot_product
- xformers.components.feedforward
- xformers.components.multi_head_dispatch
- xformers.components.positional_embedding
- xformers.components.positional_embedding.rotary
- xformers.components.positional_embedding.sine
- xformers.components.positional_embedding.vocab
- xformers.components.reversible
- xformers.ops.fmha