Metadata-Version: 2.1
Name: mqa
Version: 0.1.0
Summary: Multi Query Attention package
License: MIT
Author: kyegomez
Author-email: kyegomez@github.com
Requires-Python: >=3.8,<4.0
Classifier: License :: OSI Approved :: MIT License
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.8
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Requires-Dist: einops (>=0.3.2,<0.4.0)
Requires-Dist: flash-attn (==1.0.3.post0)
Requires-Dist: torch (>=1.10.0,<2.0.0)
Requires-Dist: triton (==2.0.0.dev20221202)
Requires-Dist: triton-pre-mlir @ git+https://github.com/vchiley/triton.git@triton_pre_mlir#subdirectory=python
