Root Mean Square Layer Normalization
RoPE position encoding via rotation
Precompute cos/sin frequency tables
Multi-query attention with shared KV heads
Expand KV heads to match query heads
Gated feedforward with Swish activation
Full Llama 3 transformer block
Complete Llama 3 model forward pass