mlx.core.distributed.Group#
- class Group#
An
mlx.core.distributed.Grouprepresents a group of independent mlx processes that can communicate.- __init__(*args, **kwargs)#
Methods
__init__(*args, **kwargs)rank(self)Get the rank of this process
size(self)Get the size of the group
split(self, color[, key])Split the group to subgroups based on the provided color.