PyPI page
Home page
Author:
Tri Dao
License:
BSD 3-Clause License
Summary:
Flash Attention CUTE (CUDA Template Engine) implementation
Latest version:
4.0.0b36
Required dependencies:
apache-tvm-ffi
|
einops
|
nvidia-cutlass-dsl
|
quack-kernels
|
torch
|
torch-c-dlpack-ext
|
typing-extensions
Optional dependencies:
nvidia-cutlass-dsl
|
pytest
|
pytest-xdist
|
ruff
Downloads last day:
36,649
Downloads last week:
202,004
Downloads last month:
859,207
7-day smoothing