Function layernorm 

pub fn layernorm(
    x: &[f32],
    gamma: &[f32],
    beta: &[f32],
    rows: usize,
    cols: usize,
    eps: f32,
) -> Vec<f32>

LayerNorm over the last dim (biased variance, like PyTorch).