mlx.core.distributed.is_available

Contents

mlx.core.distributed.is_available#

is_available(backend: str = 'any') bool#

Check if a communication backend is available.

Note, this function returns whether MLX has the capability of instantiating that distributed backend not whether it is possible to create a communication group. For that purpose one should use init(strict=True).

Parameters:

backend (str, optional) – The name of the backend to check for availability. It takes the same values as init(). Default: "any".

Returns:

Whether the distributed backend is available.

Return type:

bool