2 unstable releases

new 0.21.0-rc.1 Jan 19, 2025
0.14.0 Jul 13, 2024

#105 in Machine learning

Download history 9/week @ 2024-09-23 1/week @ 2024-09-30 2/week @ 2024-12-09

75 downloads per month

MIT/Apache

1.5MB
32K SLoC

Rust 23K SLoC // 0.0% comments C++ 7.5K SLoC // 0.0% comments Python 1K SLoC // 0.5% comments C 321 SLoC // 0.0% comments

mlx-rs

Rust bindings for Apple's mlx machine learning library.

Discord Current Crates.io Version Documentation Test Status Blaze Rust Version license

⚠️ Project is still in development - contributors welcome!



Blaze supports this project by providing ultra-fast Apple Silicon macOS Github Action Runners. Apply the discount code AI25 at checkout to enjoy 25% off your first year.

Feature Flags

  • metal - enables metal (GPU) usage in MLX
  • accelerate - enables using the accelerate framework in MLX

Versioning

For simplicity, the main crate mls-rs follows MLX’s versioning, allowing you to easily see which MLX version you’re using under the hood. The mlx-sys crate follows the versioning of mlx-c, as that is the version from which the API is generated. The mlx-macros crate uses its own versioning, as those macros are developed independently by us.

Community

If you are excited about the project or want to contribute, don't hesitate to join our Discord! We try to be as welcoming as possible to everybody from any background. We're still building this out, but you can ask your questions there!

Status

mlx-rs is currently in active development, and is not yet complete.

MSRV

The minimum supported Rust version is 1.81.0.

The MSRV is the minimum Rust version that can be used to compile each crate.

License

mlx-rs is distributed under the terms of the MIT license. See LICENSE for details. Opening a pull request is assumed to signal agreement with these licensing terms.

Dependencies

~3–11MB
~123K SLoC