david/flash-attention: flash-attention from https://github.com/Dao-AILab/flash-attention @ fa3-kvcache-gqa

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

Tri Dao 320fb59487 Update citation		il y a 6 mois
..
__init__.py	ece539abd6 Add __init__.py files to subdirectories for installation	il y a 2 ans
benchmark.py	79bd1a2d5d [LayerNorm] Implement residual + LayerNorm/RMSNorm in Triton	il y a 1 an
distributed.py	f1a73d0740 Run isort and black on python files	il y a 1 an
generation.py	320fb59487 Update citation	il y a 6 mois
pretrained.py	ef6d8c75d9 [GPT] Fix loading weights from HF hub	il y a 1 an