fn rms_norm_shift(x: &[f32], w: &[bf16], eps: f32) -> Vec<f32>
RMSNorm with the raw HF weight convention: x * inv * (1 + w).