Add sparse MLA backward (DSA top-k), sliding-window attention + attention sink, and block-sparse MLA forward/backward for SM100 - #198

Add sparse MLA backward (DSA top-k), sliding-window attention + attention sink, and block-sparse MLA forward/backward for SM100#198

kmswin1 wants to merge 3 commits into

deepseek-ai:maindeepseek-ai/FlashMLA:mainfrom

kmswin1:feat/mla-training-extensionskmswin1/FlashMLA-fork:feat/mla-training-extensionsCopy head branch name to clipboard