amorehead/jvp_flash_attention

Flash Attention Triton kernel with support for second-order derivatives

语言：Python · GitHub 星标：183 · Fork：16

在 GitHub 上查看原仓库