mirror of
https://github.com/ml-explore/mlx.git
synced 2025-06-25 18:11:15 +08:00
20 lines
394 B
ReStructuredText
20 lines
394 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
|