mlx.core.distributed.init

初始化(严格： 布尔值 = 假, 后端： str = '任何') → 组

初始化通信后端并创建全局通信组。

示例

导入 mlx.core 作为 mx

group = mx.distributed.init(backend=”ring”)

Parameters：

• strict (bool, 可选) – 如果设置为False，当mx.distributed.is_available()返回False时，它将返回一个单例组，否则会抛出运行时错误。默认值：False

• backend (str, 可选) – 选择一个特定的分布式后端进行初始化。如果设置为 any，则尝试所有可用的后端并返回第一个成功的后端。后续调用将返回第一个初始化的后端。默认值：any

Returns：

代表所有已启动进程的组。

Return type：

组