david/flash-attention

дзеркало https://github.com/Dao-AILab/flash-attention

Автор	SHA1 Опис	Дата
Tri Dao	393882bc08 [LayerNorm] Implement LN with parallel residual, support dim 8k	1 рік тому
Tri Dao	6738d9477d [LayerNorm] Implement RMS Norm	1 рік тому
Tri Dao	ae137ed17a [LayerNorm] Fuse LayerScale	2 роки тому
Tri Dao	8c6609ae1a [LayerNorm] Support all dimensions up to 6k (if divisible by 8)	2 роки тому
Tri Dao	fa6d1ce44f Add fused_dense and dropout_add_layernorm CUDA extensions	2 роки тому