Distributed Operations

int mlx_distributed_all_gather(mlx_array *res, const mlx_array x, const mlx_distributed_group group, const mlx_stream S)

int mlx_distributed_all_sum(mlx_array *res, const mlx_array x, const mlx_distributed_group group, const mlx_stream s)

int mlx_distributed_recv(mlx_array *res, const int *shape, size_t shape_num, mlx_dtype dtype, int src, const mlx_distributed_group group, const mlx_stream s)

int mlx_distributed_recv_like(mlx_array *res, const mlx_array x, int src, const mlx_distributed_group group, const mlx_stream s)

int mlx_distributed_send(mlx_array *res, const mlx_array x, int dst, const mlx_distributed_group group, const mlx_stream s)