jvp_flash_attention
★ 8
updated 9mo ago
⑂ fork
Flash Attention Triton kernel with support for second-order derivatives
No plain-English explanation yet — one is being written right now. Check back in a minute.