Instructions to use kernels-community/flash-attn2 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Kernels
How to use kernels-community/flash-attn2 with Kernels:
# !pip install kernels from kernels import get_kernel # a version (or an explicit revision) is required; see the "Files and versions" tab for the available ones kernel = get_kernel("kernels-community/flash-attn2", version=1) - Notebooks
- Google Colab
- Kaggle
Commit History
-
fix: refactor bindings and update py library for correctness
drbh876ac68
-
fix: simplify and align signatures further
drbh3099d65 -
fix: align kernel source with latest reference source
drbhc743a32
-
fix: revise bindings and wrapper typing
drbhab4cc6a
-
fix: adjust types
drbh4080f9c
-
fix: expand build combinations and include all files
drbhb0d3c12
-
feat: pass vars into fwd and include build
drbh39b4aba -
feat: include source and enable build
drbha7165c8