.coveragerc
.gitignore
.pre-commit-config.yaml
LICENSE
README.md
pyproject.toml
.github/FUNDING.yml
.github/workflows/publish.yaml
.github/workflows/test.yaml
doc/attention.png
doc/benchmark_attention.png
doc/benchmark_t5.png
doc/benchmark_t5_original.png
grouped_query_attention_pytorch/__init__.py
grouped_query_attention_pytorch/attention.py
grouped_query_attention_pytorch/t5.py
grouped_query_attention_pytorch/transformer.py
grouped_query_attention_pytorch.egg-info/PKG-INFO
grouped_query_attention_pytorch.egg-info/SOURCES.txt
grouped_query_attention_pytorch.egg-info/dependency_links.txt
grouped_query_attention_pytorch.egg-info/requires.txt
grouped_query_attention_pytorch.egg-info/top_level.txt
grouped_query_attention_pytorch/utils/__init__.py
grouped_query_attention_pytorch/utils/benchmark.py
scripts/README.md
scripts/benchmark_attention.py
scripts/benchmark_t5.py
tests/__init__.py
tests/conftest.py
tests/test_attention.py
tests/test_t5.py
tests/test_transformer.py