david/flash-attention

spiegel van https://github.com/Dao-AILab/flash-attention

Auteur	SHA1 Bericht	Datum
Tri Dao	36587c01cb [LayerNorm] Update layer_norm_linear	9 maanden geleden
Tri Dao	bdcae547c7 [LayerNorm] Don't exit early in the backward pass (fix #781)	11 maanden geleden
Tri Dao	c9861a032d [LayerNorm] Initialize mean and rstd tensor using x.device	1 jaar geleden
Tri Dao	f5b308e258 [LayerNorm] Rename layernorm.py -> layer_norm.py	1 jaar geleden