Rust Pytorch

Open-source Rust projects categorized as Pytorch

Top 6 Rust Pytorch Projects

  • burn

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

    Rust bindings for the C++ api of PyTorch.

    Project mention: One-file Rust implementation of Llama2 | | 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 (pytorch's torchlib rust binding).

  • SonarLint

    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.


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

  • bastionlab

    A simple framework for privacy-friendly data science collaboration

    Project mention: BastionLab: A privacy-by-design DataFrame library | | 2023-02-20
  • repugnant-pickle

    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:

  • modelbox

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

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

    I am building modelbox right now - 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.

Rust Pytorch related posts


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

Project Stars
1 burn 3,661
2 tch-rs 3,275
3 PERSIA 365
4 bastionlab 157
5 repugnant-pickle 15
6 modelbox 12
Free Global Payroll designed for tech teams
Building a great tech team takes more than a paycheck. Zero payroll costs, get AI-driven insights to retain best talent, and delight them with amazing local benefits. 100% free and compliant.