Function all_gather 

pub fn all_gather(x: &Array, group: Option<&Group>) -> Array

Every rank’s x concatenated along the first axis in rank order (a scalar gathers to a vector), as MLX’s all_gather.