kernels-community
/

triton-layer-norm

Model card Files Files and versions Community

triton-layer-norm

Commit History

Build

9fc83e6

danieldk HF staff commited on 3 days ago

black

15057d1

danieldk HF staff commited on 3 days ago

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

02598b2

danieldk HF staff commited on 3 days ago

Move `RMSNorm` to `layers`

02bea52

danieldk HF staff commited on 3 days ago

Build

5fdbebc

danieldk HF staff commited on 3 days ago

Add Triton-based layer norm from flash-attention

0f75957

danieldk HF staff commited on 3 days ago