david/flash-attention

mirror de https://github.com/Dao-AILab/flash-attention

Autor	SHA1 Mensagem	Data
sclarkson	1feb711f46 Fix compilation with clang on ARM64 (#1285)	2 meses atrás
Tri Dao	dff68c2b22 Add smoothing for CrossEntropyParallel, rename to CrossEntropyLoss	2 anos atrás
Tri Dao	7c9953815a Add fused cross entropy loss	2 anos atrás