Rust Machine Learning

Open-source Rust projects categorized as Machine Learning | Edit details

Top 23 Rust Machine Learning Projects

  • leaf

    Open Machine Intelligence Framework for Hackers. (GPU/CPU)

    Project mention: [D] Why does AMD do so much less work in AI than NVIDIA? | reddit.com/r/MachineLearning | 2021-11-09

    I used a lot of the dependencies behind the leaf framework which was abandoned by its authors a while back due to funding issues, as I implemented it in Rust and most bindings were maintained while the leaf framework itself wasn't anymore.

  • rust

    Rust language bindings for TensorFlow (by tensorflow)

    Project mention: Integrating machine learning models into Rust applications? | reddit.com/r/rust | 2022-03-19

    (3) You could use TensorFlow as your executor: https://github.com/tensorflow/rust

  • SonarQube

    Static code analysis for 29 languages.. Your projects are multi-language. So is SonarQube analysis. Find Bugs, Vulnerabilities, Security Hotspots, and Code Smells so you can release quality code every time. Get started analyzing your projects today for free.

  • linfa

    A Rust machine learning framework.

    Project mention: What is a FOSS which is needed but doesn't exist yet/needs contributers? | reddit.com/r/rust | 2022-02-16

    Check out smartcore and linfa. At work I was badly in need of an NMF function similar to MATLAB's one these days but not enough time to write one myself. If you're good at math and machine learning, this sounds like a task you could try tackling.

  • tch-rs

    Rust bindings for the C++ api of PyTorch.

    Project mention: [D] To what extent can Rust be used for Machine Learning? | reddit.com/r/MachineLearning | 2022-04-14

    There's a good quality wrapper for the torch library: https://github.com/LaurentMazare/tch-rs

  • tangram

    Tangram makes it easy for programmers to train, deploy, and monitor machine learning models.

    Project mention: Train a Machine Learning Model to Predict the Programming Language in a Code Snippet | dev.to | 2022-02-15

    Head over to https://www.tangram.dev and give it a try!

  • rust-bert

    Rust native ready-to-use NLP pipelines and transformer-based models (BERT, DistilBERT, GPT2,...)

    Project mention: [D] Is Rust stable/mature enough to be used for production ML? Is making Rust-based python wrappers a good choice for performance heavy uses and internal ML dependencies in 2021? | reddit.com/r/MachineLearning | 2021-12-30

    If you are using BERT models and some miscellaneous other related stuff then you should check out the rust-bert and Bert Sentence repos https://github.com/guillaume-be/rust-bert

  • juice

    The Hacker's Machine Learning Engine

    Project mention: Announcing neuronika 0.1.0, a deep learning framework in Rust | reddit.com/r/rust | 2021-06-15

    How does it compare to juice?

  • Scout APM

    Less time debugging, more time building. Scout APM allows you to find and fix performance issues with no hassle. Now with error monitoring and external services monitoring, Scout is a developer's best friend when it comes to application development.

  • rustlearn

    Machine learning crate for Rust

  • nlprule

    A fast, low-resource Natural Language Processing and Text Correction library written in Rust.

  • PERSIA

    High performance distributed framework for training deep learning recommendation models based on PyTorch.

    Project mention: Researchers Introduce ‘PERSIA’: A PyTorch-Based System for Training Large Scale Deep Learning Recommendation Models up to 100 Trillion Parameters | reddit.com/r/machinelearningnews | 2021-12-05

    Github: https://github.com/persiaml/persia

  • are-we-learning-yet

    How ready is Rust for Machine Learning?

    Project mention: [D] Is Rust stable/mature enough to be used for production ML? Is making Rust-based python wrappers a good choice for performance heavy uses and internal ML dependencies in 2021? | reddit.com/r/MachineLearning | 2021-12-30

    Hey OP, you might want to check this site out: http://arewelearningyet.com

  • nnsplit

    Semantic text segmentation. For sentence boundary detection, compound splitting and more.

  • smartcore

    SmartCore is a comprehensive library for machine learning and numerical computing. The library provides a set of tools for linear algebra, numerical computing, optimization, and enables a generic, powerful yet still efficient approach to machine learning.

    Project mention: What is a FOSS which is needed but doesn't exist yet/needs contributers? | reddit.com/r/rust | 2022-02-16

    Check out smartcore and linfa. At work I was badly in need of an NMF function similar to MATLAB's one these days but not enough time to write one myself. If you're good at math and machine learning, this sounds like a task you could try tackling.

  • autograph

    Machine Learning Library for Rust

    Project mention: Where to Learn Vulkan for parallel computation (with references to porting from CUDA) | reddit.com/r/vulkan | 2022-01-04

    I'm working on a machine learning library https://github.com/charles-r-earp/autograph implemented in Rust that uses rust-gpu to compile Rust compute shaders to spirv, and then gfx_hal to target metal and dx12. Training performance is currently about 2x slower than pytorch (cuda) on my laptop but I've made significant progress recently and I am targeting 1.5x. While rust-gpu itself has it's own restrictions, it does support inline spirv assembly, which provides direct access to operations not provided in its std lib, thus it's lower level than GLSL. For example, it should be possible to target cuda tensor cores via cooperative matrix operations (I believe Metal supports these as well but this may not be implemented in spirv-cross and certainly isn't in naga). Once I have things a bit more stabilized I'd like to provide more examples, like porting from cuda / opencl, but I'm still figuring out patterns like how to work with 16 and 8 bit types in a nice and portable way.

  • L2

    l2 is a fast, Pytorch-style Tensor+Autograd library written in Rust

  • rsrl

    A fast, safe and easy to use reinforcement learning framework in Rust.

    Project mention: Understanding Rust through machine learning code. | reddit.com/r/rust | 2022-01-18

    rsrl/transforms.rs at master · tspooner/rsrl (github.com) (thanks to the authors.)

  • tractjs

    Run ONNX and TensorFlow inference in the browser.

    Project mention: Run WASM, a client side Python runtime | news.ycombinator.com | 2021-09-26

    Tensorflow (and by extension Keras) offload most of the actual work to C++ or C, so having those compile to WebAssembly would (I imagine) be a herculean effort.

    Instead, The TF team maintains TFJS, which can run on WebAssembly[0].

    There are also tractjs[1], and onnyxjs[2], both of which allow you to run (most) ONNX models (which is an open standard for specifying ML models) using WebAssembly and WebGL(only onnyxjs supports WebGL). A bunch of frameworks (caffe, pytorch, TF) support exporting to/importing from ONNX.

    [0] https://blog.tensorflow.org/2020/03/introducing-webassembly-...

    [1] https://github.com/bminixhofer/tractjs

    [2] https://github.com/microsoft/onnxjs

  • txtai.rs

    Build AI-powered semantic search applications in Rust

    Project mention: Transform tabular data with composable workflows | dev.to | 2021-11-10

    Next we'll execute the workflow. txtai has API bindings for JavaScript, Java, Rust and Golang. But to keep things simple, we'll just run the commands via cURL.

  • bhtsne

    Parallel Barnes-Hut t-SNE implementation written in Rust.

    Project mention: bhtsne 0.5.0, now 5.6x faster on a 4 core machine, plus a summary of my Rust journey (so far) | reddit.com/r/rust | 2021-10-15

    bhtsne now supports parallelism, custom data types and custom user defined metrics.

  • CNTK

    Wrapper around Microsoft CNTK library

  • corgi

    A neural network, and tensor dynamic automatic differentiation implementation for Rust. (by patricksongzy)

  • speak

    Talk with your machine in this minimalistic Rust crate!

    Project mention: Introducing Speak, the machine-learning chatbot written in Rust | reddit.com/r/rust | 2022-05-08

    Repository

  • pair_adjacent_violators

    An implementation of the Pair Adjacent Violators algorithm for isotonic regression in Rust

NOTE: The open source projects on this list are ordered by number of github stars. The number of mentions indicates repo mentiontions in the last 12 Months or since we started tracking (Dec 2020). The latest post mention was on 2022-05-08.

Rust Machine Learning related posts

Index

What are some of the best open-source Machine Learning projects in Rust? This list will help you:

Project Stars
1 leaf 5,475
2 rust 3,640
3 linfa 1,709
4 tch-rs 1,614
5 tangram 1,310
6 rust-bert 930
7 juice 891
8 rustlearn 526
9 nlprule 407
10 PERSIA 294
11 are-we-learning-yet 253
12 nnsplit 242
13 smartcore 241
14 autograph 203
15 L2 152
16 rsrl 124
17 tractjs 66
18 txtai.rs 62
19 bhtsne 43
20 CNTK 21
21 corgi 16
22 speak 12
23 pair_adjacent_violators 3
Find remote jobs at our new job board 99remotejobs.com. There are 8 new remote jobs listed recently.
Are you hiring? Post a new remote job listing for free.
SaaSHub - Software Alternatives and Reviews
SaaSHub helps you find the best software and product alternatives
www.saashub.com