#mlx #machine-learning #deep-learning

mlx-rs

Unofficial rust wrapper for Apple's mlx machine learning library

1 unstable release

0.14.0 Jul 13, 2024

#139 in Machine learning

MIT/Apache

665KB
15K SLoC

Rust 12K SLoC // 0.0% comments C++ 2.5K SLoC // 0.0% comments Python 420 SLoC // 0.1% comments C 148 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.

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

~1.3–4MB
~81K SLoC