preloader

FlexAttention: The Flexibility of PyTorch with the Performance of FlashAttention

@misc{He_Guessous_Liang_Dong_2024, title={https://pytorch.org/blog/flexattention/}, url={https://pytorch.org/blog/flexattention/}, journal={FlexAttention: The Flexibility of PyTorch with the Performance of FlashAttention}, publisher={PYTorch}, author={He, Horace and Guessous, Driss and Liang, Yanbo and Dong, Joy}, year={2024}, month={Aug}}