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
-
Build uploaded using `kernels`.
Verified signature98af189
-
Build uploaded using `kernels`.
Verified signatureb0129a2 -
Build uploaded using `kernels`.
Verified signature20f621e
-
Build uploaded using `kernels`.
Verified signature172e232
-
Build uploaded using `kernels`.
Verified signature35815ed
-
Fix kernel module name
380f85b
-
Build uploaded using `kernels`.
Verified signaturebe86063
-
Build uploaded using `kernels`.
Verified signature046d51d
-
fix: bump build for torch 2.8
drbh90b3e94
-
feat: bump build
drbh09eec95
-
feat: bump builds
drbh809bfb7
-
Build (aarch64)
f849035 -
Build
b58ed97
-
fix: update library for all ops
drbhb833fce -
Update build/torch26-cxx98-cu124-x86_64-linux/flash_attn/__init__.py
Verified signature79e0916 -
feat: bump build
David Holtzdd2a3b2
-
feat: pass vars into fwd and include build
drbh39b4aba