Top 8 Rust Pytorch Projects

burn

1 11 9,411 9.8 Rust

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

Project mention: Burn: The Future of Deep Learning in Rust | dev.to | 2024-12-26

Burn is an emerging deep learning framework written in pure Rust that aims to provide a flexible, efficient, and safe environment for building and training neural networks. With its modular design and strong type system, Burn represents a significant step forward in bringing deep learning to the Rust ecosystem.
SaaSHub

www.saashub.com featured

SaaSHub - Software Alternatives and Reviews. SaaSHub helps you find the best software and product alternatives
tch-rs

2 37 4,483 7.6 Rust

Rust bindings for the C++ api of PyTorch.
PERSIA

3 3 398 2.8 Rust

High performance distributed framework for training deep learning recommendation models based on PyTorch.
rust-mlops-template

4 2 335 5.6 Rust

A work in progress to build out solutions in Rust for MLOPs
bastionlab

5 3 170 5.1 Rust

A simple framework for privacy-friendly data science collaboration
wildbg

6 1 26 7.4 Rust

Backgammon engine based on neural networks
repugnant-pickle

7 1 25 3.0 Rust

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

8 1 13 1.3 Rust

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

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).

Rust Pytorch discussion

Rust Pytorch related posts

Tch-Rs

1 project | news.ycombinator.com | 7 Dec 2023
Llama2.rs: One-file Rust implementation of Llama2

3 projects | news.ycombinator.com | 5 Aug 2023
[D] MNIST-like dataset in SVG format

2 projects | /r/MachineLearning | 9 Jul 2023
llm: a Rust crate/CLI for CPU inference of LLMs, including LLaMA, GPT-NeoX, GPT-J and more

7 projects | /r/rust | 9 May 2023
Simply explained: How does GPT work?

1 project | news.ycombinator.com | 6 Apr 2023
BastionLab: A privacy-by-design DataFrame library

1 project | news.ycombinator.com | 20 Feb 2023
Will I ever need python again if I learn rust other than for AI stuff?

1 project | /r/Python | 4 Feb 2023
A note from our sponsor - SaaSHub
www.saashub.com | 16 Jan 2025

SaaSHub helps you find the best software and product alternatives Learn more →

Index

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

#	Project	Stars
1	burn	9,411
2	tch-rs	4,483
3	PERSIA	398
4	rust-mlops-template	335
5	bastionlab	170
6	wildbg	26
7	repugnant-pickle	25
8	modelbox	13

Rust Pytorch

Top 8 Rust Pytorch Projects

Rust Pytorch discussion

Rust Pytorch related posts

Tch-Rs

Llama2.rs: One-file Rust implementation of Llama2

[D] MNIST-like dataset in SVG format

llm: a Rust crate/CLI for CPU inference of LLMs, including LLaMA, GPT-NeoX, GPT-J and more

Simply explained: How does GPT work?

BastionLab: A privacy-by-design DataFrame library

Will I ever need python again if I learn rust other than for AI stuff?

Index