kernels-community
/

triton-layer-norm

Model card Files Files and versions Community

triton-layer-norm / torch-ext /triton_layer_norm

Ctrl+K

Ctrl+K

3 contributors

History: 1 commit

danieldk's picture

danieldk HF Staff

Add Triton-based layer norm from flash-attention

0f75957 about 2 months ago

__init__.py

161 Bytes

Add Triton-based layer norm from flash-attention about 2 months ago
layer_norm.py

35.7 kB

Add Triton-based layer norm from flash-attention about 2 months ago