GitHub repository: simudt/Tri-RMSNorm
Primary language: Python
Description: Efficient kernel for RMS normalization with fused operations, includes both forward and backward passes, compatibility with PyTorch.