Back to Mlx

Distributed Communication

docs/src/python/distributed.rst

0.31.2425 B
Original Source

.. _distributed:

.. currentmodule:: mlx.core.distributed

Distributed Communication

MLX provides a distributed communication package using MPI. The MPI library is loaded at runtime; if MPI is available then distributed communication is also made available.

.. autosummary:: :toctree: _autosummary

Group
is_available
init
all_sum
all_gather
send
recv
recv_like