pub fn rms_norm(x: &mut [f32], w: &[bf16], eps: f32)
RMSNorm: x_i * w_i / sqrt(mean(x^2) + eps). Computed in f32.