daking's picture vchiley's picture
add flash_attn_triton.py (#9)
8f0448e