Link: https://github.com/unslothai/unsloth/blob/main/unsloth/kernels/rms_layernorm.py
Author: Daniel Han-Chen & the Unsloth team
Tags: Normalization
Description:
- RMSNorm forward
- RMSNorm backward
Triton Version: ?
Other Notes:
-
Id in triton index: 0005