Machine Learning in Rust

Discover cutting-edge open source machine learning projects built with Rust. Explore fast, efficient, and reliable ML tools that leverage Rust's performance and safety. Boost your data science workflow with these innovative Rust-powered machine learning libraries.

Projects

  1. burn icon

    #1burn

    Burn: A dynamic deep learning framework in Rust offering flexibility, efficiency, and portability across multiple backends. Optimized for performance with features like automatic kernel fusion and asynchronous execution.

    9k•458•Rust•Apache-2.0
    Screenshot of burn website

  2. #2candle

    Candle is a minimalist ML framework for Rust, focusing on performance and ease of use. It enables serverless inference, removes Python from production workloads, and supports GPU acceleration.

    16k•979•Rust•Apache-2.0
  3. ort icon

    #3ort

    Accelerate machine learning inference and training on CPU & GPU with this Rust wrapper for ONNX Runtime. Seamlessly integrate powerful ML capabilities into your Rust projects.

    985•105•Rust•Apache-2.0
    Screenshot of ort website

  4. #4tract

    Tract is a powerful Neural Network inference toolkit that optimizes and executes ONNX and NNEF models efficiently across various devices, from microcontrollers to Raspberry Pis.

    2k•216•Rust
  5. rust-bert icon

    #5rust-bert

    Rust-native state-of-the-art Natural Language Processing models and pipelines, offering high-performance implementations of popular NLP tasks.

    2k•217•Rust•Apache-2.0
    Screenshot of rust-bert website

  6. #6linfa

    Linfa: A comprehensive machine learning toolkit for Rust, offering a wide range of algorithms and preprocessing tools for everyday ML tasks. Inspired by scikit-learn, it aims to provide a robust foundation for building ML applications.

    3k•252•Rust•Apache-2.0

  7. #7rust

    Idiomatic Rust bindings for TensorFlow, enabling seamless integration of machine learning capabilities into Rust projects with GPU support and flexible compilation options.

    5k•422•Rust•Apache-2.0