david/flash-attention: flash-attention from https://github.com/Dao-AILab/flash-attention @ 6807b1ea37254102f5d0e3f3e3e37949cc548a89

-ын хуулбар https://github.com/Dao-AILab/flash-attention

Tri Dao c7f32a8409 [CrossEntropy] Support precomputed LSE		3 сар өмнө
..
test_cross_entropy.py	c7f32a8409 [CrossEntropy] Support precomputed LSE	3 сар өмнө
test_cross_entropy_parallel.py	c7f32a8409 [CrossEntropy] Support precomputed LSE	3 сар өмнө