A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
BSD 3-Clause "New" or "Revised" License
8.2k
stars
1.36k
forks
source link
A FasterRMSNorm implementation (based on FasterLayerNorm) #1688
Open
Njuapp opened 1 year ago
A FasterRMSNorm implementation (similar to FasterLayerNorm).