Rust Tensor Frameworks

PyTorch-style tensor libraries and autograd systems implemented in Rust, emphasizing safety and performance. Does NOT include data versioning, compression utilities, or reasoning systems unrelated to core tensor operations.

There are 66 rust tensor frameworks tracked. 4 score above 50 (established tier). The highest-rated is EnzymeAD/Enzyme at 62/100 with 1,558 stars. 3 of the top 10 are actively maintained.

Get all 66 projects as JSON

curl "https://pt-edge.onrender.com/api/v1/datasets/quality?domain=ml-frameworks&subcategory=rust-tensor-frameworks&limit=20"

Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.

# Framework Score Tier
1 EnzymeAD/Enzyme

High-performance automatic differentiation of LLVM and MLIR.

62
Established
2 LaurentMazare/tch-rs

Rust bindings for the C++ api of PyTorch.

58
Established
3 Oxen-AI/Oxen

Lightning fast data version control system for structured and unstructured...

55
Established
4 strake-data/strake

The Data Layer for AI. A high-performance federated SQL engine that gives AI...

52
Established
5 SunDoge/dlpark

A Rust Library for High-Performance Tensor Exchange with Python

49
Emerging
6 TheMesocarp/koho

Full spectrum sheaf neural network over arbitrary CW complexes.

42
Emerging
7 Dicklesworthstone/frankentorch

Memory-safe clean-room Rust reimplementation of PyTorch with deterministic...

39
Emerging
8 TrainPioneers/Tensor-Frame

A high-performance, PyTorch-like tensor library for Rust with support for...

38
Emerging
9 Photoroom/datago

A natively parallel dataloader for Python, written in Rust. Serving data at...

38
Emerging
10 cool-japan/torsh

ToRSh (Tensor Operations in Rust with Sharding) is a PyTorch-compatible deep...

38
Emerging
11 Dicklesworthstone/frankenjax

Clean-room Rust reimplementation of JAX transform semantics (jit, grad,...

38
Emerging
12 octoflow-lang/octoflow

GPU-Native Programming Language. 4.5 MB binary. Any GPU. Zero dependencies.

38
Emerging
13 zk4x/zyx

Tensor library for machine learning

37
Emerging
14 m96-chan/candle-miotts

Pure Rust MioTTS inference — RTF 0.21 on a single GPU, no Python needed

37
Emerging
15 cool-japan/tenflowers

A pure Rust implementation of TensorFlow, providing a full-featured machine...

37
Emerging
16 frjnn/bhtsne

Parallel Barnes-Hut t-SNE implementation written in Rust.

37
Emerging
17 chelsea0x3b/dfdx

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

36
Emerging
18 eaplatanios/ryft

A Rust Framework for Tracing, Automatic Differentiation, and Just-In-Time Compilation

36
Emerging
19 charles-r-earp/autograph

A machine learning library for Rust.

35
Emerging
20 miniex/hodu

A machine learning library built with user convenience at its core

35
Emerging
21 Genius-apple/RusTorch

RusTorch is a production-grade deep learning framework re-imagined in Rust....

34
Emerging
22 maciejkula/wyrm

Autodifferentiation package in Rust.

34
Emerging
23 raskr/rust-autograd

Tensors and differentiable operations (like TensorFlow) in Rust

34
Emerging
24 bilal2vec/L2

l2 is a fast, Pytorch-style Tensor+Autograd library written in Rust

33
Emerging
25 c0dearm/mushin

Computational graphs with reverse automatic differentation in the GPU

29
Experimental
26 Jianqoq/Hpt

A high performance N-dimensional array library for Rust

28
Experimental
27 fsprojects/Furnace

Production-grade ML - F# power & precision guiding Torch performance

28
Experimental
28 VasanthakumarV/einops

Simplistic API for deep learning tensor operations

28
Experimental
29 Entrolution/echidna

A high-performance automatic differentiation library for Rust

28
Experimental
30 sheaf-lang/sheaf

A functional framework for machine learning. Inspired by Clojure, compiled...

27
Experimental
31 NextStat/nextstat.io

Universal differentiable statistical inference engine (Rust + GPU)....

26
Experimental
32 gussmith23/glenside

A pure, low-level tensor program representation enabling tensor program...

24
Experimental
33 ramsyana/RustTensor

A learning-focused, high-performance tensor computation library built from...

23
Experimental
34 sablokrep/datathon

rust crate for optimized deep learning

22
Experimental
35 torchforge-rs/torchforge-viz

Lightweight, well-maintained training visualizer for Rust ML pipelines.

22
Experimental
36 torchforge-rs/torchforge-data

Zero-copy, streaming data pipeline for edge-native machine learning in Rust.

22
Experimental
37 xpressgarageplus/RusTorch

Build and run deep learning models in Rust with a PyTorch-like API that...

22
Experimental
38 arclabs561/propago

Graph neural network primitives and training loops.

22
Experimental
39 arclabs561/structops

Differentiable dynamic programming operators (Soft-DTW, soft shortest path)...

22
Experimental
40 lightsofapollo/safetensors-streaming

Drop-in Rust replacement for safetensors that loads model weights faster

22
Experimental
41 syntheticore/microtensor

Automatic differentiation for tensor operations

21
Experimental
42 tzervas/unsloth-rs

Memory-optimized GPU kernels for LLM fine-tuning in Rust (2-5x speedup,...

21
Experimental
43 bokutotu/zenu

A Deep Learning framework with very few dependencies, Written in Rust

21
Experimental
44 BobMcDear/rusten

A minimal, didactic implementation of tensors in Rust.

20
Experimental
45 sablokrep/graphsage

rust graphsage

19
Experimental
46 mattdeeds/irontensor

Minimal Rust tensor library for training LLMs on Apple Silicon with Metal...

19
Experimental
47 PaytonWebber/zerograd

A small, educational library for neural networks, implemented from zero in pure Rust.

18
Experimental
48 GregorLueg/node2vec-rs

Rust-based implementations of node2vec via Burn or CPU-accelerated version

17
Experimental
49 wesleyscholl/ZipGraph

⚡ Blazing-fast graph processing engine in Rust with ML-powered optimization....

16
Experimental
50 springyworks/candlekos

Fork focusing on new tensor ops for Candle: FFT and Scan, plus an...

15
Experimental
51 elftausend/gradients

Deep Learning library written in Rust (OpenCL, CUDA & CPU)

15
Experimental
52 james0248/tangent

An autodiff engine in Rust with JAX-style tracing, forward/reverse-mode...

15
Experimental
53 Selithrarion/ai-engine-deep-dive

deconstructing llms, transformers, autograd, oonx, gguf, gpu kernels, cubecl

14
Experimental
54 arjunr00/ml-rust-cuda

A set of GPU-accelerated machine learning algorithms and functions written...

14
Experimental
55 MDResearch/Mady

🌲 Ahead-of-time Static Macro-gen Automatic Differentiation. A little bit...

14
Experimental
56 SunDoge/fastdata

Fast Data Pipeline with TFRecord and io-uring

14
Experimental
57 asukaminato0721/d2l.ai-rs

use candle to implement some of the d2l.ai

13
Experimental
58 NickLucche/autograd.rs

Simple Deep Learning library in Rust based on ndarray.

12
Experimental
59 wilhelmagren/tensorust

Tensor library in Rust.

12
Experimental
60 vilhelmbergsoe/nn

(WIP) Simple Deep Learning Framework and Auto Differentiation Engine in Rust

11
Experimental
61 matiasbattocchia/simple-grad

Automatic differentiation in Rust for educational purposes. Autograd /...

11
Experimental
62 uddp/eph

eph is a data-oriented high-performance revision control system designed for...

11
Experimental
63 rafal-draws/chrust

chrust (torCH + RUST) is a hybrid machine learning web application for music...

11
Experimental
64 hatellezp/rustoner

The rustoner project aims to create a description logic reasoner in rust....

10
Experimental
65 saivishwak/flowtorch

Safe and Performant Scientific Computing Library

10
Experimental
66 njord-rs/njord-pipeline

A highly versatile and feature-rich data pipeline library for Rust

10
Experimental