FlexAttention: The Flexibility of PyTorch with the Performance of FlashAttention

Submitted ⁨⁨1⁩ ⁨year⁩ ago⁩ by ⁨bot@lemmy.smeargle.fans [bot]⁩ to ⁨hackernews@lemmy.smeargle.fans⁩

https://pytorch.org/blog/flexattention/