Function mlx_distributed_recv 

pub unsafe extern "C" fn mlx_distributed_recv(
    res: *mut mlx_array,
    shape: *const c_int,
    shape_num: usize,
    dtype: mlx_dtype,
    src: c_int,
    group: mlx_distributed_group,
    s: mlx_stream,
) -> c_int