kernels-community
/

triton-layer-norm

Model card Files Files and versions

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

Commit History

Add top-level `layer_norm` wrapper with docs

80b5db1

danieldk HF Staff commited on Jun 20

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