kernels-community
/

triton-layer-norm

Model card Files Files and versions

triton-layer-norm

Commit History

Generate README

9b61b27

danieldk HF Staff commited on Jun 20

Update build

9c1f92e

danieldk HF Staff commited on Jun 20

Update flake

ba5e096

danieldk HF Staff commited on Jun 20

Update build.toml

042f669

danieldk HF Staff commited on Jun 20

Add top-level `layer_norm` wrapper with docs

80b5db1

danieldk HF Staff commited on Jun 20

fix: adjust typo

491036b
verified

drbh HF Staff commited on May 6

feat: add tag for hfjob build

ce36e5f
verified

drbh HF Staff commited on May 6

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

4a07eb0

danieldk HF Staff commited on Mar 14

feat: update to include rev in kernel for reproducible symbols

1e347f0

drbh commited on Mar 25

Build

9fc83e6

danieldk HF Staff commited on Mar 12

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

Build

5fdbebc

danieldk HF Staff commited on Mar 12

Add Triton-based layer norm from flash-attention

0f75957

danieldk HF Staff commited on Mar 12