david/flash-attention

Mirror von https://github.com/Dao-AILab/flash-attention

Autor	SHA1 Nachricht	Datum
Tri Dao	393882bc08 [LayerNorm] Implement LN with parallel residual, support dim 8k	vor 1 Jahr
Tri Dao	6738d9477d [LayerNorm] Implement RMS Norm	vor 1 Jahr
Tri Dao	ae137ed17a [LayerNorm] Fuse LayerScale	vor 2 Jahren
Tri Dao	8c6609ae1a [LayerNorm] Support all dimensions up to 6k (if divisible by 8)	vor 2 Jahren
Tri Dao	fa6d1ce44f Add fused_dense and dropout_add_layernorm CUDA extensions	vor 2 Jahren