vchiley's picture
add flash_attn_triton.py
a118b32