Wheelodex » Projects » memory-efficient-attention

memory-efficient-attention

View on PyPI — Reverse Dependencies (0)

0.1.3

memory_efficient_attention-0.1.3-py3-none-any.whl

Wheel Details

Project:	memory-efficient-attention
Version:	0.1.3
Filename:	memory_efficient_attention-0.1.3-py3-none-any.whl
Download:	[link]
Size:	8469
MD5:	`14e3c4f64de992b9ed44bb5bf5f6501e`
SHA256:	`33535d1f9b369ded0de9763e363fa5143ed331ab08cf785197d567a8167b219e`
Uploaded:	2022-03-27 13:09:13 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

Metadata-Version:	2.1
Name:	memory-efficient-attention
Version:	0.1.3
Summary:	Memory Efficient Attention (O(sqrt(n)) for Jax and PyTorch
Author:	Amin Rezaei
Author-Email:	AminRezaei0x443[at]gmail.com
Home-Page:	https://github.com/AminRezaei0x443/memory-efficient-attention
License:	MIT
Keywords:	attention,pytorch,jax
Requires-Dist:	numpy
Requires-Dist:	jax; extra == "jax"
Requires-Dist:	jax; extra == "testing"
Requires-Dist:	torch; extra == "testing"
Requires-Dist:	flax; extra == "testing"
Requires-Dist:	torch; extra == "torch"
Provides-Extra:	jax
Provides-Extra:	testing
Provides-Extra:	torch

[No description]

WHEEL

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

RECORD

Path	Digest	Size
memory_efficient_attention/__init__.py	sha256=WaUO6LC65y0j8hLBTj1w0j0rrHLtLGT6b3pvuFhjnQo	197
memory_efficient_attention/attention_jax.py	sha256=qirAIDfYBK8A9i6dYYeIfiMrb_CibRtbqk3eREuoIyg	8997
memory_efficient_attention/attention_torch.py	sha256=xwSOvSver9vwoLSVpGS7LGF9pzfO9-jTB52LblbkDE4	8759
memory_efficient_attention/utils.py	sha256=nwgrSxuTxdIj3zwrs-l7O-AxSTbWbdqRzIj7mXZ1V_A	734
test/__init__.py	sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU	0
test/test_computation.py	sha256=yqxE7FkpH3st5O9jOQKu5tsaARb1Y4oMoV9pJ46KcuQ	7935
memory_efficient_attention-0.1.3.dist-info/METADATA	sha256=Ck5uaZmNTAFz8zAvngbuAhgbX33lOEzS4Kgn4dK3xE4	630
memory_efficient_attention-0.1.3.dist-info/WHEEL	sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA	92
memory_efficient_attention-0.1.3.dist-info/top_level.txt	sha256=BSAlxPWFRgPpLo83IdiYYyNBibj2ykdi1helGfyhmjY	32
memory_efficient_attention-0.1.3.dist-info/RECORD	—	—

top_level.txt

memory_efficient_attention

test