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

peilaus alkaen https://github.com/Dao-AILab/flash-attention

Tri Dao c7f32a8409 [CrossEntropy] Support precomputed LSE		3 kuukautta sitten
..
__init__.py	ece539abd6 Add __init__.py files to subdirectories for installation	2 vuotta sitten
cross_entropy.py	c7f32a8409 [CrossEntropy] Support precomputed LSE	3 kuukautta sitten