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: 6 commits

ror's picture

ror

Update autotune configuration to avoid crash on AMD devices

da878ae verified 27 days ago

__init__.py

173 Bytes

Move `RMSNorm` to `layers` 3 months ago
layer_norm.py

36.4 kB

Update autotune configuration to avoid crash on AMD devices 27 days ago
layers.py

524 Bytes

Make `RMSNorm` layer pure, rename to `LlamaRMSNorm` 3 months ago