search

oxideai / mlx-rs

Unofficial Rust bindings to Apple's mlx framework
https://oxideai.github.io/mlx-rs/
Apache License 2.0
63 stars 5 forks source link

Support distributed ops #104

Open dcvz opened 1 month ago

dcvz commented 1 month ago

In 15.2 distributed ops were introduced: https://github.com/ml-explore/mlx/blob/969337345ffd34159884c8bf124dbb6eeaa902bb/docs/src/usage/distributed.rst#L29