PyPI page
Home page
Author:
None
License:
BSD 3-Clause License
Copyright (c) 2025, the respective contributors, as shown by the AUTHORS file.
All rights reserved.
Redistribution and use in source and ...
Summary:
Flash Sparse Attention: Fast and Memory-Efficient Trainable Sparse Attention
Latest version:
2.0.1
Required dependencies:
triton
Optional dependencies:
apache-tvm-ffi
|
einops
|
mkdocs
|
mkdocs-material
|
mkdocstrings
|
numpy
|
nvidia-cutlass-dsl
|
pymdown-extensions
|
pytest
|
pytest-benchmark
|
quack-kernels
|
tabulate
|
torch-c-dlpack-ext
|
transformers
|
typing_extensions
Downloads last day:
12
Downloads last week:
36
Downloads last month:
163