PyPI page
Home page
Author:
Phil Wang
License:
MIT License JVP Flash Attention (jvp_flash_attention) Copyright (c) 2025, The Regents of the University of California, through Lawrence Berkeley National Laboratory (subject to receipt of any require...
Summary:
Flash Attention Triton kernel with support for second-order derivatives (Lucidrains fork)
Latest version:
0.0.1
Required dependencies:
torch
Optional dependencies:
numpy
|
pytest
Downloads last day:
0
Downloads last week:
21
Downloads last month:
0
7-day smoothing