kernels-community
/

triton-layer-norm

Model card Files Files and versions Community

triton-layer-norm / torch-ext /triton_layer_norm /layers.py

Commit History

Add top-level `layer_norm` wrapper with docs

80b5db1

danieldk HF Staff commited on 1 day ago

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

4a07eb0

danieldk HF Staff commited on Mar 14

Move `RMSNorm` to `layers`

02bea52

danieldk HF Staff commited on Mar 12