Function normal 

pub fn normal(
    shape: &[usize],
    dtype: DType,
    loc: Option<f64>,
    scale: Option<f64>,
    key: Option<&Array>,
) -> Result<Array>

Normal samples (mean loc, standard deviation scale) of shape and floating-point dtype, as MLX: √2 · erfinv(u) for u uniform in (-1, 1).