piwheels

flash-attention-jax

Flash Attention - in Jax

Installation

In a virtualenv (see these instructions if you need to create one):

pip3 install flash-attention-jax

Dependencies

• einops
• jax

Releases

Version	Released	Bookworm Python 3.11	Trixie Python 3.13	Files
0.3.1	2023-07-18
flash_attention_jax-0.3.1-py3-none-any.whl (10 KB)
0.3.0	2023-06-18
flash_attention_jax-0.3.0-py3-none-any.whl (10 KB)
0.2.0	2022-11-01
flash_attention_jax-0.2.0-py3-none-any.whl (10 KB)
0.1.0	2022-09-29
flash_attention_jax-0.1.0-py3-none-any.whl (10 KB)
0.0.10	2022-09-21
flash_attention_jax-0.0.10-py3-none-any.whl (10 KB)
0.0.9	2022-08-21
flash_attention_jax-0.0.9-py3-none-any.whl (9 KB)
0.0.8	2022-08-20
flash_attention_jax-0.0.8-py3-none-any.whl (9 KB)
0.0.7	2022-07-27
flash_attention_jax-0.0.7-py3-none-any.whl (9 KB)
0.0.6	2022-07-23
flash_attention_jax-0.0.6-py3-none-any.whl (8 KB)
0.0.5	2022-07-22
flash_attention_jax-0.0.5-py3-none-any.whl (8 KB)
0.0.4	2022-07-22
flash_attention_jax-0.0.4-py3-none-any.whl (8 KB)
0.0.2	2022-07-22
flash_attention_jax-0.0.2-py3-none-any.whl (6 KB)
0.0.1	2022-07-22
flash_attention_jax-0.0.1-py3-none-any.whl (6 KB)

Issues with this package?

• Search issues for this package
• Package or version missing? Open a new issue
• Something else? Open a new issue