mlx/docs/src/python/distributed.rst

.. _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
Add docs for the distributed namespace (#1184) 2024-06-07 02:37:00 +08:00			`.. _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`
Adds send/recv ops in distributed (#1366) 2024-08-27 14:01:37 +08:00			`send`
			`recv`
			`recv_like`