i404788/Tri-RMSNorm
Efficient kernel for RMS normalization with fused operations, includes both forward and backward passes, compatibility with PyTorch.
PythonApache-2.0
Watchers
No one’s watching this repository yet.
Efficient kernel for RMS normalization with fused operations, includes both forward and backward passes, compatibility with PyTorch.
PythonApache-2.0
No one’s watching this repository yet.