tensorplay.nn.attention.list_flash_attention_impls

tensorplay.nn.attention.list_flash_attention_impls() → list[str]

Return the names of all available flash attention implementations.