piwheels

jvp-flash-attention

Flash Attention Triton kernel with support for second-order derivatives

Installation

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

pip3 install jvp-flash-attention

Dependencies

• torchvision
• torch

Releases

Version	Released	Bullseye Python 3.9	Bookworm Python 3.11	Trixie Python 3.13	Files
0.14.0	2026-03-10
jvp_flash_attention-0.14.0-py3-none-any.whl (27 KB)
0.13.0	2026-02-24
jvp_flash_attention-0.13.0-py3-none-any.whl (26 KB)
0.12.0	2026-02-04
jvp_flash_attention-0.12.0-py3-none-any.whl (26 KB)
0.11.0	2025-12-21
jvp_flash_attention-0.11.0-py3-none-any.whl (26 KB)
0.10.0	2025-10-21
jvp_flash_attention-0.10.0-py3-none-any.whl (25 KB)
0.0.9	2025-10-07
jvp_flash_attention-0.0.9-py3-none-any.whl (25 KB)
0.0.8	2025-10-01
jvp_flash_attention-0.0.8-py3-none-any.whl (25 KB)
0.0.7	2025-09-30
jvp_flash_attention-0.0.7-py3-none-any.whl (25 KB)
0.0.6	2025-09-17
jvp_flash_attention-0.0.6-py3-none-any.whl (25 KB)
0.0.5	2025-09-16
jvp_flash_attention-0.0.5-py3-none-any.whl (25 KB)
0.0.4	2025-09-15
jvp_flash_attention-0.0.4-py3-none-any.whl (24 KB)
0.0.3	2025-09-08
jvp_flash_attention-0.0.3-py3-none-any.whl (21 KB)
0.0.2	2025-09-06
jvp_flash_attention-0.0.2-py3-none-any.whl (21 KB)
0.0.1	2025-09-03

Issues with this package?

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