SaaSHub helps you find the best software and product alternatives Learn more →
Top 11 Rust Pytorch Projects
-
burn
Burn is a next generation tensor library and Deep Learning Framework that doesn't compromise on flexibility, efficiency and portability.
Project mention: Show HN: Demucs music stem separator rewritten in Rust – runs in the browser | news.ycombinator.com | 2026-03-03- *DAW plugin* — VST3/CLAP plugin for macOS with a native SwiftUI UI. Load a track, separate it, drag stems directly into your DAW timeline, or play as a MIDI instrument with solo / faders.
The core inference library is built on [Burn](https://burn.dev), a Rust deep learning framework. The same `demucs-core` crate compiles to both native and `wasm32-unknown-unknown` — the only thing that changes is the GPU backend.
Model weights are F16 safetensors hosted on Hugging Face and downloaded / cached automatically on first use on all platforms. Three variants: standard 4-stem (84 MB), 6-stem with guitar/piano (84 MB), and a fine-tuned bag-of-4-models for best quality (333 MB).
The existing implementations I found online were mostly wrappers around the original Python implementation, and not very portable -- the model works remarkably well and I wanted to be able to quickly create samples / remixes without leaving the DAW or my browser. Right now the implementation is pretty MacOS heavy, as that's what I'm testing with, but all of the building blocks for other platforms are ready to build on. I want this to grow to be a general utility for music producers, not just "works on my machine."
It was a fun first foray into DSP and the state of the art of ML over WASM, with lots of help from Claude!
-
SaaSHub
SaaSHub - Software Alternatives and Reviews. SaaSHub helps you find the best software and product alternatives
-
-
PERSIA
High performance distributed framework for training deep learning recommendation models based on PyTorch.
-
-
-
-
repugnant-pickle
Low effort scraping Python's pickle format in Rust. It is to complete pickle parsing as BeautifulSoup was to complete HTML parsing.
-
-
modelbox
An extensible machine learning model store and model transformation and distribution service
-
-
tpu-doc
tpu-doc is a zero-dependency diagnostic binary for Google Cloud TPU environments that instantly validates hardware health, discovers software stack configurations, and provides AI-powered log analysis to eliminate expensive debugging downtime.
Project mention: Show HN: TPU-doc – A zero-dependency diagnostic tool for Google Cloud TPU health | news.ycombinator.com | 2026-01-06
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
-
CUDA-oxide: Nvidia's official Rust to CUDA compiler
-
Agent Orchestration for the Timid
-
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?
-
A note from our sponsor - SaaSHub
www.saashub.com | 21 Jul 2026
Index
What are some of the best open-source Pytorch projects in Rust? This list will help you:
| # | Project | Stars |
|---|---|---|
| 1 | burn | 15,594 |
| 2 | tch-rs | 5,449 |
| 3 | PERSIA | 414 |
| 4 | rust-mlops-template | 380 |
| 5 | bastionlab | 175 |
| 6 | wildbg | 53 |
| 7 | repugnant-pickle | 27 |
| 8 | ferric-micrograd | 15 |
| 9 | modelbox | 14 |
| 10 | diffai | 3 |
| 11 | tpu-doc | 2 |
Sponsored
SaaSHub - Software Alternatives and Reviews
SaaSHub helps you find the best software and product alternatives
www.saashub.com