Function mlx_count_nonzero 

pub unsafe extern "C" fn mlx_count_nonzero(
    res: *mut mlx_array,
    a: mlx_array,
    keepdims: bool,
    s: mlx_stream,
) -> c_int