Implement RMSNorm from scratch in PyTorch

Transformer Architecture Internals2 / 7