Function split 

pub fn split(a: &Array, parts: usize, axis: isize) -> Result<Vec<Array>>

a split into parts equal parts along axis (MLX’s split with a count); views.