Top 6 Rust Pytorch Projects

burn

31 3,661 9.2 Rust

Burn - A Flexible and Comprehensive Deep Learning Framework in Rust

Project mention: Why Rust Is the Optimal Choice for Deep Learning, and How to Start Your Journey with the Burn Deep Learning Framework | dev.to | 2023-09-07

The comprehensive, open-source deep learning framework in Rust, Burn, has recently undergone significant advancements in its latest release, highlighted by the addition of The Burn Book 🔥. There has never been a better moment to embark on your deep learning journey with Rust, as this book will guide you through your initial project, providing extensive explanations and links to relevant resources.
tch-rs

19 3,275 7.1 Rust

Rust bindings for the C++ api of PyTorch.

Project mention: Llama2.rs: One-file Rust implementation of Llama2 | news.ycombinator.com | 2023-08-05

I wanted to do something like this but then I would miss on proper CUDA acceleration and lose performance compared to using torchlib.
I wrote a forgettable llama implementation for https://github.com/LaurentMazare/tch-rs (pytorch's torchlib rust binding).
SonarLint

www.sonarlint.org
sponsored

Clean code begins in your IDE with SonarLint. Up your coding game and discover issues early. SonarLint is a free plugin that helps you find & fix bugs and security issues from the moment you start writing code. Install from your favorite IDE marketplace today.
PERSIA

0 365 0.0 Rust

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

3 157 10.0 Rust

A simple framework for privacy-friendly data science collaboration

Project mention: BastionLab: A privacy-by-design DataFrame library | news.ycombinator.com | 2023-02-20
repugnant-pickle

1 15 10.0 Rust

Low effort scraping Python's pickle format in Rust. It is to complete pickle parsing as BeautifulSoup was to complete HTML parsing.

Project mention: Introducing repugnant-pickle, a crate for scraping Python Pickle files in a basic way. Notable, it can deal with (some) PyTorch model files. | /r/rust | 2023-04-06

The repo has a detailed README: https://github.com/KerfuffleV2/repugnant-pickle
modelbox

1 12 10.0 Rust

An extensible machine learning model store and model transformation and distribution service

Project mention: Using Rust at a Startup: A Cautionary Tale | news.ycombinator.com | 2022-11-22

I am building modelbox right now - https://github.com/tensorland/modelbox I began building this in Rust while also learning the language. It became quickly very complex as I started introducing streams in async traits. I think in a few years things will get a lot better as more people use Rust for building web services, but I had to go back to Go to be more productive and ship this thing out. I loved how the compiler was forcing me to think harder about life times and such.

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 2023-09-07.