kernels-community
/

flash-attn

Model card Files Files and versions Community

flash-attn / torch-ext /flash_attn

Ctrl+K

Ctrl+K

3 contributors

History: 4 commits

drbh

fix: refactor bindings and update py library for correctness

876ac68 7 days ago

layers
fix: refactor bindings and update py library for correctness 7 days ago
ops
fix: refactor bindings and update py library for correctness 7 days ago
__init__.py

14.2 kB

fix: refactor bindings and update py library for correctness 7 days ago
bert_padding.py

9.93 kB

fix: refactor bindings and update py library for correctness 7 days ago
flash_attn_interface.py

60.7 kB

fix: refactor bindings and update py library for correctness 7 days ago