daking's picture vchiley's picture
add flash_attn_triton.py (#34)
aeea751