← All questions
LLM FoundationsMedium· Normalization· Inference Efficiency

Why do modern LLMs use RMSNorm instead of LayerNorm?

RMSNorm is preferred in modern LLMs primarily due to memory-bandwidth efficiency. Normalization operations often account for approximately 25% of total runtime despite contributing less than 0.2% of total FLOPs. By switching to RMSNorm, models achieve significant speedups without sacrificing performance, a trend observed across multiple major model families.

Code playground (Python)