Wheelodex » Projects » grouped-query-attention-pytorch

grouped-query-attention-pytorch

0.2.0

grouped_query_attention_pytorch-0.2.0-py3-none-any.whl

Wheel Details

Project:	grouped-query-attention-pytorch
Version:	0.2.0
Filename:	grouped_query_attention_pytorch-0.2.0-py3-none-any.whl
Download:	[link]
Size:	526563
MD5:	`d99f7cd6dfedc8e21afd1135d80ef140`
SHA256:	`38df32b96c2540106e8c1a5bf2b56b82b2de08a25eead1b877dadf97d71677be`
Uploaded:	2023-08-03 00:46:20 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

Metadata-Version:	2.1
Name:	grouped-query-attention-pytorch
Version:	0.2.0
Summary:	grouped-query-attention-pytorch
Author-Email:	Frank Odom <frank.odom.iii[at]gmail.com>
License:	MIT
Classifier:	Programming Language :: Python :: 3
Requires-Python:	>=3.8
Requires-Dist:	einops (~=0.6.0)
Requires-Dist:	torch (>=1.8.0)
Requires-Dist:	torchscale (~=0.2.0)
Requires-Dist:	sentencepiece; extra == "t5"
Requires-Dist:	transformers (>=4.5.0); extra == "t5"
Requires-Dist:	black; extra == "test"
Requires-Dist:	kaleido; extra == "test"
Requires-Dist:	mypy; extra == "test"
Requires-Dist:	plotly; extra == "test"
Requires-Dist:	pre-commit; extra == "test"
Requires-Dist:	pytest; extra == "test"
Requires-Dist:	pytest-cov; extra == "test"
Requires-Dist:	ruff; extra == "test"
Requires-Dist:	xformers (==0.0.20); extra == "test"
Provides-Extra:	t5
Provides-Extra:	test
Description-Content-Type:	text/markdown
License-File:	LICENSE

[Description omitted; length: 7757 characters]

WHEEL

Wheel-Version:	1.0
Generator:	bdist_wheel (0.41.0)
Root-Is-Purelib:	true
Tag:	py3-none-any

RECORD

Path	Digest	Size
doc/attention.png	sha256=d9YRB0jQB80bLL4h0ARYBYAFiA7rG4O3t0LmXJDcGqM	335190
doc/benchmark_attention.png	sha256=toY7-jJfmQtX3RErt0gNf6DKmsud3N73qXjq8VOk5jA	37835
doc/benchmark_t5.png	sha256=J4GnUDH84AmC6T4lWeu6DPCnlHhLEHN8-CUYDQZ3dDE	37657
doc/benchmark_t5_original.png	sha256=qMN2woqEj_ha5ILHva-zTIG4Hn6t2ZHIVMrGnuJiiaU	118342
grouped_query_attention_pytorch/__init__.py	sha256=HxfQexYh_kiqrgr6vZrJKh7fxpEPm74HqFkgz0JfPro	394
grouped_query_attention_pytorch/attention.py	sha256=2Lkg1ClIIVEeT_8jD2vT26BCgQ9hJV_mOQz-FNuRZko	12512
grouped_query_attention_pytorch/t5.py	sha256=WFxNH0g2zPW1ZSGjyP9E4M1IG5h7_naY1IBmEVpXDJ0	13858
grouped_query_attention_pytorch/transformer.py	sha256=xbb4thvwO_bTcMkEOB3sCixK9xd-_IbGhX1eHWKSJ6o	12863
grouped_query_attention_pytorch/utils/__init__.py	sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU	0
grouped_query_attention_pytorch/utils/benchmark.py	sha256=2pxotv2PHQxSfQ2yKpt63u_Jc7OkwgpIc5cmv93ycXw	1732
scripts/README.md	sha256=uHmq28tfNm-Kj6UbFq1va-SD-jlRri9wcjWug5Vqsyo	2340
scripts/benchmark_attention.py	sha256=u7FDiKrcSMTfzN355lPWqGqm-X1_W2gLiq_9vgkonJU	5207
scripts/benchmark_t5.py	sha256=CLRJtnLVEEKblOgpaapnvTyi_4wkNCRnFTHRsODxmT8	9082
grouped_query_attention_pytorch-0.2.0.dist-info/LICENSE	sha256=hRXuyjlIpCesw-fD9ONCbyH2AZxw9AKL2IMx3EcCOEs	1067
grouped_query_attention_pytorch-0.2.0.dist-info/METADATA	sha256=QhpK_lMi0TC52ZnLJpS5fPx5cMwVHKmmktLr5_jb_ic	8690
grouped_query_attention_pytorch-0.2.0.dist-info/WHEEL	sha256=AtBG6SXL3KF_v0NxLf0ehyVOh0cold-JbJYXNGorC6Q	92
grouped_query_attention_pytorch-0.2.0.dist-info/top_level.txt	sha256=2U_uv68-EqxEABp3Ypm8NVZBnJwOc17pLbRy7ZqqreU	44
grouped_query_attention_pytorch-0.2.0.dist-info/RECORD	—	—

top_level.txt

doc

grouped_query_attention_pytorch

scripts