squall/flash-attention
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
081c2b012a
flash-attention/flash_attn/ops/triton
History
Tri Dao 6f6e9a9aaf [FusedDense] Enable sqrelu activation in FusedMLP
2023-04-13 15:29:32 -07:00
..
k_activations.py Add GPT and ViT models 2022-11-13 22:30:23 -08:00
linear.py Add GPT and ViT models 2022-11-13 22:30:23 -08:00
mlp.py [FusedDense] Enable sqrelu activation in FusedMLP 2023-04-13 15:29:32 -07:00