daking's picture vchiley's picture
add flash_attn_triton.py (#28)
f89f0f8