MLX
Loading...
Searching...
No Matches
distributed.h File Reference
#include <memory>
#include "mlx/array.h"

Go to the source code of this file.

Classes

struct  mlx::core::distributed::Group
 A distributed::Group represents a group of independent mlx processes that can communicate. More...
 

Namespaces

namespace  mlx
 
namespace  mlx::core
 
namespace  mlx::core::distributed
 

Functions

bool mlx::core::distributed::is_available ()
 
Group mlx::core::distributed::init (bool strict=false)
 Initialize the distributed backend and return the group containing all discoverable processes.