Wheelodex » Projects » flash-attention-jax

flash-attention-jax

View on PyPI — Reverse Dependencies (0)

0.3.1

flash_attention_jax-0.3.1-py3-none-any.whl

Wheel Details

Project:	flash-attention-jax
Version:	0.3.1
Filename:	flash_attention_jax-0.3.1-py3-none-any.whl
Download:	[link]
Size:	10066
MD5:	`cc637cc4799f0d7986f103a815562217`
SHA256:	`20dc2f01c671517afaac83483329bf0ccdb89b20f817bb6ea5be09bd8e5c950c`
Uploaded:	2023-07-18 02:45:16 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

Metadata-Version:	2.1
Name:	flash-attention-jax
Version:	0.3.1
Summary:	Flash Attention - in Jax
Author:	Phil Wang
Author-Email:	lucidrains[at]gmail.com
Home-Page:	https://github.com/lucidrains/flash-attention-jax
License:	MIT
Keywords:	artificial intelligence,deep learning,transformers,attention mechanism,jax
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:	einops
Requires-Dist:	jax (>=0.2.20)
Description-Content-Type:	text/markdown
License-File:	LICENSE

[No description]

WHEEL

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

RECORD

Path	Digest	Size
flash_attention_jax/__init__.py	sha256=GrELmkaEsupZzlva42CWxToppOLNSuDsDLj5Tkp0Yk8	494
flash_attention_jax/attention.py	sha256=0s5uPhciTYoD5QZXKXhcw2Rhw8V_ZyBoHQMNxNPuWO8	1424
flash_attention_jax/causal_flash_attention.py	sha256=51xhMkCjczFdDX08I-0A98egMW7OCvaNtusR63s5zaU	7448
flash_attention_jax/cosine_sim_flash_attention.py	sha256=YTM-bQyixUAbg-4zAco59ZVzUgpfZyW7moAjdvpehaU	5822
flash_attention_jax/flash_attention.py	sha256=l8W1VImleSoCRcNWRlBMa77nyw7uS95HCYz-KSfE1vQ	6597
flash_attention_jax/rabe_attention.py	sha256=cYsrYWJ1Ky95F6UzuSt5UMoqZNfxlyvSfpez-cp2sO8	2174
flash_attention_jax/utils.py	sha256=3IrqPqh2pdRMynMZTlVexLauL6FhdgRkl3t4jAmhzwA	1348
flash_attention_jax-0.3.1.dist-info/LICENSE	sha256=xZDkKtpHE2TPCAeqKe1fjdpKernl1YW-d01j_1ltkAU	1066
flash_attention_jax-0.3.1.dist-info/METADATA	sha256=J-HfOIxvG64Rk7ysc-QTtx4yPqUluozeMKryXYdKmoc	683
flash_attention_jax-0.3.1.dist-info/WHEEL	sha256=pkctZYzUS4AYVn6dJ-7367OJZivF2e8RA9b_ZBjif18	92
flash_attention_jax-0.3.1.dist-info/top_level.txt	sha256=2AFkWEHfehP0VI2LBmSwXIaudOW407Xsi8Pd-tTbjak	20
flash_attention_jax-0.3.1.dist-info/RECORD	—	—

top_level.txt

flash_attention_jax