The shape of the array in a particular dimension.
@ffi.Native<ffi.Int Function(mlx_array, ffi.Int)>() external int mlx_array_dim(mlx_array arr, int dim);