Wheelodex » Projects » ring-attention-pytorch

ring-attention-pytorch

View on PyPI — Reverse Dependencies (1)

0.4.1 ring_attention_pytorch-0.4.1-py3-none-any.whl

Wheel Details

Project: ring-attention-pytorch
Version: 0.4.1
Filename: ring_attention_pytorch-0.4.1-py3-none-any.whl
Download: [link]
Size: 23117
MD5: 1c094c3e556900b8d26c6af445bf53cd
SHA256: 5bc20fef31526692084ce7886ddb222389d9183e98d9aa9d6bf635f8477d2cec
Uploaded: 2024-04-20 15:25:55 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

Metadata-Version: 2.1
Name: ring-attention-pytorch
Version: 0.4.1
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.7.0)
Requires-Dist: torch (>=2.0)
Description-Content-Type: text/markdown
License-File: LICENSE
[No description]

WHEEL

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

RECORD

Path Digest Size
ring_attention_pytorch/__init__.py sha256=Y4ZCU3CKAZ0HyJNrOdGCpMpEZwQA1BLCBZaFC6Lj6as 388
ring_attention_pytorch/distributed.py sha256=VedAo7ZIQPEJIyhxIHOzevaOn9dq2P4DO394bhRdxMc 3079
ring_attention_pytorch/ring.py sha256=jHO5AqNzLBLRVX39XB20sEvDXtMq60Rkh7drBXFzWfw 4000
ring_attention_pytorch/ring_attention.py sha256=4qmFQJOX4nuim2Zzg7GuiW_zpQU4YPIpvzj_JMamjgs 19870
ring_attention_pytorch/ring_flash_attention.py sha256=XkmyiqbUl5r0PmqRVkpk7OIqTvYmfS_AWhknbr93wF8 13077
ring_attention_pytorch/ring_flash_attention_cuda.py sha256=3kZ5W-_DhO76RQEI7MknNtSjUbwpuqvL9OfLLoFvgpM 10956
ring_attention_pytorch/triton_flash_attn.py sha256=5jewrsUnYNeuY191snVgE259QIrD9HIr5mfuEo6_VXA 33543
ring_attention_pytorch-0.4.1.dist-info/LICENSE sha256=E7RGS7kpJIStk5za_-4DVhWEAamf65EU0CNML25mq4c 1066
ring_attention_pytorch-0.4.1.dist-info/METADATA sha256=qdpBhjHJa4dkhHF3iz1mz8BMyNpkbBLUBw2HMeyjqRo 703
ring_attention_pytorch-0.4.1.dist-info/WHEEL sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ 92
ring_attention_pytorch-0.4.1.dist-info/top_level.txt sha256=WM7rK5Jnbp-zxzvLUg5Hn2VjbQzSj9Ilkb9qFG3kUd8 23
ring_attention_pytorch-0.4.1.dist-info/RECORD — —

top_level.txt

ring_attention_pytorch