Function normal 

pub fn normal<'a>(
    shape: impl IntoShape,
    dtype: DType,
    loc: f64,
    scale: f64,
    key: impl Into<Option<&'a Array>>,
) -> Array

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