mirror of
https://github.com/ml-explore/mlx.git
synced 2025-07-15 21:21:16 +08:00
23 lines
426 B
ReStructuredText
23 lines
426 B
ReStructuredText
.. _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
|