Wheelodex » Projects » ring-attention-pytorch

ring-attention-pytorch

View on PyPI — Reverse Dependencies (2)

0.5.19 ring_attention_pytorch-0.5.19-py3-none-any.whl

Wheel Details

Project: ring-attention-pytorch
Version: 0.5.19
Filename: ring_attention_pytorch-0.5.19-py3-none-any.whl
Download: [link]
Size: 27472
MD5: a1255feecfd0c1fb561cdf2d6146b478
SHA256: 81723bf510a47776b493bb4521528681525484c4c4fe722d6b3a6dc0ab717913
Uploaded: 2024-10-25 15:23:20 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

Metadata-Version: 2.1
Name: ring-attention-pytorch
Version: 0.5.19
Summary: Ring Attention - Pytorch
Author: Phil Wang
Author-Email: lucidrains[at]gmail.com
Home-Page: https://github.com/lucidrains/ring-attention-pytorch
License: MIT
Keywords: artificial intelligence,deep learning,distributed attention
Classifier: Development Status :: 4 - Beta
Classifier: Intended Audience :: Developers
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Classifier: License :: OSI Approved :: MIT License
Classifier: Programming Language :: Python :: 3.6
Requires-Dist: beartype
Requires-Dist: einops (>=0.8.0)
Requires-Dist: jaxtyping
Requires-Dist: torch (>=2.0)
Description-Content-Type: text/markdown
License-File: LICENSE
[No description]

WHEEL

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

RECORD

Path Digest Size
ring_attention_pytorch/__init__.py sha256=MJ1L_O4jiT9gkD332NnMYc1SFY8obBC4PU6n63DITG8 468
ring_attention_pytorch/distributed.py sha256=p5mKtaAmN03dxOKMPySvdY6r5l_xJFBs8IiQPS9PySQ 3533
ring_attention_pytorch/ring.py sha256=NOpPd2_xN-99gxCifc-WoFW6-ZNG3RRQZdANcoBCp8I 3735
ring_attention_pytorch/ring_attention.py sha256=cha_XGz3S1aIoe5UVgBbcDek23FKznu2eXu4nV6HJK0 20276
ring_attention_pytorch/ring_flash_attention.py sha256=vhxbVnd_6j8m1txjkDdXJQ8D8_7FZiwPLuzVZyWCRv4 14097
ring_attention_pytorch/ring_flash_attention_cuda.py sha256=scy_EGiKkM2XzK9Qg22QVsY2G4zc-l6QXOA-QGbxoas 11478
ring_attention_pytorch/tensor_typing.py sha256=2aUeDq34TmWeurXF9E93Gj_be3houMePK1ZsJuV3-d4 447
ring_attention_pytorch/tree_attn_decoding.py sha256=D0be1HGLX_0wRxmRAaRlDRhXjAQk9l3V_O39-P5Zag8 2957
ring_attention_pytorch/triton_flash_attn.py sha256=Hpp6v4giVBfBP7zdJQbT4d5vNB6zlX6qPz1XREDBZAY 35413
ring_attention_pytorch/zig_zag_attention.py sha256=RN21ZELccQSf1DiAqpvnkJIfSBnDpv7dPy58DU4jYEA 3604
ring_attention_pytorch-0.5.19.dist-info/LICENSE sha256=E7RGS7kpJIStk5za_-4DVhWEAamf65EU0CNML25mq4c 1066
ring_attention_pytorch-0.5.19.dist-info/METADATA sha256=Fa1TmwAn8q93kGsyxHLAzBUOM_QlNqvh7Uj87ym-5CE 729
ring_attention_pytorch-0.5.19.dist-info/WHEEL sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA 91
ring_attention_pytorch-0.5.19.dist-info/top_level.txt sha256=WM7rK5Jnbp-zxzvLUg5Hn2VjbQzSj9Ilkb9qFG3kUd8 23
ring_attention_pytorch-0.5.19.dist-info/RECORD — —

top_level.txt

ring_attention_pytorch