ringattention

View on PyPIReverse Dependencies (0)

0.1.2 ringattention-0.1.2-py3-none-any.whl

Wheel Details

Project: ringattention
Version: 0.1.2
Filename: ringattention-0.1.2-py3-none-any.whl
Download: [link]
Size: 19627
MD5: 43b5fab86634def3318763490f62f422
SHA256: 89cc587c01a6e49c286dc1163fba69375236b8f7f06d1b1d06536b80a68c81fa
Uploaded: 2024-08-12 19:40:13 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: ringattention
Version: 0.1.2
Summary: RingAttention for Transformers with Arbitrarily Large Context.
Project-Url: Homepage, https://github.com/forhaoliu/ringattention
License: Apache-2.0
Classifier: Development Status :: 3 - Alpha
Classifier: Intended Audience :: Developers
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Programming Language :: Python :: 3 :: Only
Requires-Python: >=3.10
Requires-Dist: numpy
Requires-Dist: jax (>=0.4.29)
Requires-Dist: einops
Requires-Dist: flax
License-File: LICENSE
[No description]

WHEEL

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

RECORD

Path Digest Size
ringattention/__init__.py sha256=wp9i4JuDVE37T_utVYZYP9qcXdbYY-DR0NwyuscNw3s 1295
ringattention/ringattention_inference.py sha256=Ud0bBnjG_QRQYN3Vv2YeuYwC-HBxeYX6z-9mbtxb1_E 3987
ringattention/ringattention_jax.py sha256=3CtxHALMVfyeHLP19jFBKndddSoPlRy1dqSSBzy7h10 18086
ringattention/ringattention_pallas_gpu.py sha256=4uUNq4anMsm_r_htDhVRAbLC4GTdo1c1xd1s5KBNVeg 553
ringattention/ringattention_pallas_tpu.py sha256=GJSnFAKT7hpcgZbEk--FuEG-VUWZGGCofx9DBxlSKko 55061
ringattention-0.1.2.dist-info/LICENSE sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ 11357
ringattention-0.1.2.dist-info/METADATA sha256=FkWoulLMwz1cOk1i8KnRB1GOoxjoyYU6onCbZe09qOk 559
ringattention-0.1.2.dist-info/WHEEL sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs 91
ringattention-0.1.2.dist-info/top_level.txt sha256=CfLqlQkBTNXmVZFG7KYgH9BII07MM3p2EQHh42pwGK8 14
ringattention-0.1.2.dist-info/RECORD

top_level.txt

ringattention