david/flash-attention: flash-attention from https://github.com/Dao-AILab/flash-attention @ ipiszy/used_q

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

Curtis "Fjord" Hawthorne d8aacc510c return z_loss (#768)		há 1 ano atrás
..
test_cross_entropy.py	d8aacc510c return z_loss (#768)	há 1 ano atrás
test_cross_entropy_parallel.py	08124c8f9c [CrossEntropy] Implement logit_scale option	há 1 ano atrás