replicate/flash-mla

1---2library_name: kernels3license: mit4---5 6> [!CAUTION]7> Starting from September 13, 2026, we will be removing the "model" type repositories of kernels (e.g., kernels-community/flash-attn3). Make sure you're using a latest version of kernels. If you face any disruption, please report them here: https://github.com/huggingface/kernels/issues/new.8 9This is the repository card of kernels-community/flash-mla that has been pushed on the Hub. It was built to be used with the [`kernels` library](https://github.com/huggingface/kernels). This card was automatically generated.10 11## How to use12 13```python14# make sure `kernels` is installed: `pip install -U kernels`15from kernels import get_kernel16 17kernel_module = get_kernel("kernels-community/flash-mla")18__version__ = kernel_module.__version__19 20__version__(...)21```22 23## Available functions24- `__version__`25- `FlashMLASchedMeta`26- `get_mla_metadata`27- `flash_mla_with_kvcache`28- `flash_attn_varlen_func`29- `flash_attn_varlen_qkvpacked_func`30- `flash_attn_varlen_kvpacked_func`31- `flash_mla_sparse_fwd`32 33## Benchmarks34 35Benchmarking script is available for this kernel. Run `kernels benchmark kernels-community/flash-mla`.36