mx.split(a, indices, axis): split at the given boundary offsets along axis into N+1 contiguous slices (mirrors numpy/mlx split-by-indices).
axis
mx.split(a, indices, axis): split at the given boundary offsets along
axisinto N+1 contiguous slices (mirrors numpy/mlx split-by-indices).