Function rms_norm 

pub fn rms_norm(x: &[f32], w: &[f32], eps: f32) -> Vec<f32>

Vanilla RMSNorm: x * w / sqrt(mean(x^2) + eps).