Morph MORPH
SIGN IN SIGN UP
tejas / flash-attention

Fast and memory-efficient exact attention

0 0 208 Python

[Cute,Bwd,Sm100] Simplify mma by using functools.partial

T
Tri Dao committed
7b17cd8b693661097d5586358db63d5607e0efea
Parent: c0c8c2d