├── .gitignore ├── LICENSE ├── README.md ├── mqa ├── __init__.py ├── flash_attn_triton.py └── main.py ├── pyproject.toml └── requirements.txt /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/MultiQueryAttention/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/MultiQueryAttention/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/MultiQueryAttention/HEAD/README.md -------------------------------------------------------------------------------- /mqa/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/MultiQueryAttention/HEAD/mqa/__init__.py -------------------------------------------------------------------------------- /mqa/flash_attn_triton.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/MultiQueryAttention/HEAD/mqa/flash_attn_triton.py -------------------------------------------------------------------------------- /mqa/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/MultiQueryAttention/HEAD/mqa/main.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/MultiQueryAttention/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/MultiQueryAttention/HEAD/requirements.txt --------------------------------------------------------------------------------