16.3 kB
axjns's picture
Add Triton flash attention (2-3x over AOTriton, 31x less peak memory)
b3e26c5 verified