Rust machine learning libraries

17 libraries, 9 updated this year

Rust's machine learning projects focus on fast, dependency-light inference. Hugging Face's candle is a minimal tensor framework for running models on CPU, CUDA and Metal; burn is a full framework that also trains, with swappable backends. Hugging Face's tokenizers library is written in Rust and powers its Python package.

17 libraries

candlehuggingface/candle

Candle is a minimalist ML framework for Rust with a focus on performance (including GPU support) and ease of use.

General

21.2k updated 10d ago

burntracel-ai/burn

Burn is a new comprehensive dynamic Deep Learning Framework built using Rust with extreme flexibility, compute efficiency and portability as its primary goals

Deep learning

16.2k updated today

huggingface/tokenizershuggingface/tokenizers

Fast State-of-the-Art Tokenizers optimized for Research and Production

Deep learningNLP

11.2k updated 1d ago

shimmyMichael-A-Kuykendall/shimmy

Python-free Rust inference server for NLP models with OpenAI API compatibility and hot model swapping.

LLMsNLP

6k updated 5d ago

leafautumnai/leaf Deprecated

Open source framework for machine intelligence, sharing concepts from TensorFlow and Caffe. Available under the MIT license.

General

5.5k updated 2y ago

tch-rsLaurentMazare/tch-rs

Rust bindings for the C++ API of PyTorch

Deep learning

5.5k updated 2mo ago

linfarust-ml/linfa

A comprehensive toolkit to build Machine Learning applications with Rust

General

4.8k updated 2mo ago

rust-bertguillaume-be/rust-bert

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

Deep learningNLP

3.1k updated 9mo ago

dfdxcoreylowman/dfdx

Deep learning in Rust, with shape checked tensors and neural networks

Deep learning

1.9k updated 2y ago

rusty-machineAtheMathmo/rusty-machine Archived

A pure-rust machine learning library.

General

1.3k archived

smartcoresmartcorelib/smartcore

"The Most Advanced Machine Learning Library In Rust."

General

969 updated today

rustlearnmaciejkula/rustlearn

A machine learning framework featuring logistic regression, support vector machines, decision trees and random forests.

General

647 updated 5y ago

deltadelta-rs/delta

An open source machine learning framework in Rust Δ

General

414 updated 1y ago

RustNNjackm321/RustNN Deprecated

RustNN is a feedforward neural network library.

GeneralDeep learning

339 updated 8y ago

deeplearn-rstedsta/deeplearn-rs Archived

Deeplearn-rs provides simple networks that use matrix multiplication, addition, and ReLU under the MIT license.

GeneralDeep learning

210 archived

RusticSOMavinashshenoy97/RusticSOM

A Rust library for Self Organising Maps (SOM).

General

36 updated 4y ago

shimmytokMichael-A-Kuykendall/shimmytok

Pure-Rust tokenizer for GGUF models, compatible with llama.cpp tokenization.

LLMsNLP

25 updated 6d ago