kernels-community
/

flash-attn

Model card Files Files and versions

flash-attn / torch-ext /flash_attn /ops /triton

Ctrl+K

Ctrl+K

3 contributors

History: 1 commit

drbh

fix: refactor bindings and update py library for correctness

876ac68 about 1 month ago

__init__.py

1 Bytes

fix: refactor bindings and update py library for correctness about 1 month ago
cross_entropy.py

12.9 kB

fix: refactor bindings and update py library for correctness about 1 month ago
k_activations.py

4.03 kB

fix: refactor bindings and update py library for correctness about 1 month ago
layer_norm.py

42.6 kB

fix: refactor bindings and update py library for correctness about 1 month ago
linear.py

20.8 kB

fix: refactor bindings and update py library for correctness about 1 month ago
mlp.py

6.08 kB

fix: refactor bindings and update py library for correctness about 1 month ago
rotary.py

7.09 kB

fix: refactor bindings and update py library for correctness about 1 month ago