pub fn layer_norm(
x: &Array,
weight: Option<&Array>,
bias: Option<&Array>,
eps: f32,
) -> Result<Array>Expand description
Layer normalisation over the last axis, (x - mean) / sqrt(var + eps) · weight + bias,
as MLX’s fast.layer_norm: statistics in float32, the result in the promoted type.