kernels-community
/

triton-layer-norm

Model card Files Files and versions

triton-layer-norm / torch-ext /triton_layer_norm

Commit History

Update autotune configuration to avoid crash on AMD devices

da878ae
verified

ror HF Staff commited on May 23

Make `RMSNorm` layer pure, rename to `LlamaRMSNorm`

4a07eb0

danieldk HF Staff commited on Mar 14

black

15057d1

danieldk HF Staff commited on Mar 12

Use `torch.amp` rather than deprecated `torch.cuda.amp`

02598b2

danieldk HF Staff commited on Mar 12

Move `RMSNorm` to `layers`

02bea52

danieldk HF Staff commited on Mar 12

Add Triton-based layer norm from flash-attention

0f75957

danieldk HF Staff commited on Mar 12