Function split_at 

pub fn split_at(a: &Array, indices: &[usize], axis: isize) -> Result<Vec<Array>>

a split along axis before each of the (non-decreasing) positions indices, as MLX’s split with indices: indices.len() + 1 views.