Hugging Face's logo

kernels-community
/
triton-layer-norm

Kernels
Model card Files
xet
 Community
2
triton-layer-norm

Commit History

Update tag
5ebc83a
verified

danieldk HF Staff commited on

Build
229047c

danieldk HF Staff commited on

Sync with upstream, add tests
bbbdefe

danieldk HF Staff commited on

Generate README
9b61b27

danieldk HF Staff commited on

Update build
9c1f92e

danieldk HF Staff commited on

Update flake
ba5e096

danieldk HF Staff commited on

Update build.toml
042f669

danieldk HF Staff commited on

Add top-level `layer_norm` wrapper with docs
80b5db1

danieldk HF Staff commited on

fix: adjust typo
491036b
verified

drbh HF Staff commited on

feat: add tag for hfjob build
ce36e5f
verified

drbh HF Staff commited on

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

danieldk HF Staff commited on

feat: update to include rev in kernel for reproducible symbols
1e347f0

drbh commited on

Build
9fc83e6

danieldk HF Staff commited on

black
15057d1

danieldk HF Staff commited on

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

danieldk HF Staff commited on

Move `RMSNorm` to `layers`
02bea52

danieldk HF Staff commited on

Build
5fdbebc

danieldk HF Staff commited on

Add Triton-based layer norm from flash-attention
0f75957

danieldk HF Staff commited on