PyPI page
Author:
Driss Guessous
Summary:
Helpful tools and examples for working with flex-attention
Latest version:
0.0.17
Required dependencies:
torch
Optional dependencies:
apache-tvm-ffi
|
cuda-python
|
flash-attn-4
|
jsonargparse
|
marimo
|
matplotlib
|
mkdocs-macros-plugin
|
mkdocs-material
|
mkdocs-redirects
|
mkdocstrings
|
nvidia-cutlass-dsl
|
packaging
|
prek
|
pytest
|
pytest-xdist
|
ruff
|
tabulate
|
typer
Downloads last day:
679
Downloads last week:
4,226
Downloads last month:
18,286
7-day smoothing