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 65 rust tensor frameworks tracked. 4 score above 50 (established tier). The highest-rated is EnzymeAD/Enzyme at 66/100 with 1,558 stars. 3 of the top 10 are actively maintained.

Get all 65 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	Stars	Language
1	EnzymeAD/Enzyme High-performance automatic differentiation of LLVM and MLIR.	66	Established	1,558	LLVM
2	Oxen-AI/Oxen Lightning fast data version control system for structured and unstructured...	59	Established	1,117	Rust
3	LaurentMazare/tch-rs Rust bindings for the C++ api of PyTorch.	58	Established	5,316	Rust
4	SunDoge/dlpark A Rust Library for High-Performance Tensor Exchange with Python	53	Established	52	Rust
5	TheMesocarp/koho Full spectrum sheaf neural network over arbitrary CW complexes.	49	Emerging	16	Rust
6	Photoroom/datago A natively parallel dataloader for Python, written in Rust. Serving data at...	45	Emerging	127	Rust
7	cool-japan/torsh ToRSh (Tensor Operations in Rust with Sharding) is a PyTorch-compatible deep...	44	Emerging	17	Rust
8	frjnn/bhtsne Parallel Barnes-Hut t-SNE implementation written in Rust.	44	Emerging	89	Rust
9	TrainPioneers/Tensor-Frame A high-performance, PyTorch-like tensor library for Rust with support for...	44	Emerging	2	Rust
10	chelsea0x3b/dfdx Deep learning in Rust, with shape checked tensors and neural networks	43	Emerging	1,896	Rust
11	cool-japan/tenflowers A pure Rust implementation of TensorFlow, providing a full-featured machine...	43	Emerging	4	Rust
12	charles-r-earp/autograph A machine learning library for Rust.	42	Emerging	335	Rust
13	raskr/rust-autograd Tensors and differentiable operations (like TensorFlow) in Rust	41	Emerging	500	Rust
14	zk4x/zyx Tensor library for machine learning	41	Emerging	26	Rust
15	miniex/hodu A machine learning library built with user convenience at its core	41	Emerging	108	Rust
16	maciejkula/wyrm Autodifferentiation package in Rust.	41	Emerging	179	Rust
17	bilal2vec/L2 l2 is a fast, Pytorch-style Tensor+Autograd library written in Rust	40	Emerging	197	Rust
18	Dicklesworthstone/frankentorch Memory-safe clean-room Rust reimplementation of PyTorch with deterministic...	38	Emerging	6	Rust
19	octoflow-lang/octoflow GPU-Native Programming Language. 4.5 MB binary. Any GPU. Zero dependencies.	37	Emerging	5	—
20	c0dearm/mushin Computational graphs with reverse automatic differentation in the GPU	36	Emerging	397	Rust
21	m96-chan/candle-miotts Pure Rust MioTTS inference — RTF 0.21 on a single GPU, no Python needed	36	Emerging	1	Rust
22	VasanthakumarV/einops Simplistic API for deep learning tensor operations	35	Emerging	35	Rust
23	Jianqoq/Hpt A high performance N-dimensional array library for Rust	35	Emerging	45	Rust
24	eaplatanios/ryft A Rust Framework for Tracing, Automatic Differentiation, and Just-In-Time Compilation	35	Emerging	10	Rust
25	fsprojects/Furnace Production-grade ML - F# power & precision guiding Torch performance	35	Emerging	58	F#
26	Genius-apple/RusTorch RusTorch is a production-grade deep learning framework re-imagined in Rust....	33	Emerging	85	Rust
27	sheaf-lang/sheaf A functional framework for machine learning. Inspired by Clojure, compiled...	31	Emerging	11	Rust
28	gussmith23/glenside A pure, low-level tensor program representation enabling tensor program...	31	Emerging	73	Rust
29	Entrolution/echidna A high-performance automatic differentiation library for Rust	31	Emerging	6	Rust
30	ramsyana/RustTensor A learning-focused, high-performance tensor computation library built from...	29	Experimental	70	Rust
31	syntheticore/microtensor Automatic differentiation for tensor operations	28	Experimental	6	Rust
32	bokutotu/zenu A Deep Learning framework with very few dependencies, Written in Rust	28	Experimental	64	Rust
33	BobMcDear/rusten A minimal, didactic implementation of tensors in Rust.	27	Experimental	13	Rust
34	PaytonWebber/zerograd A small, educational library for neural networks, implemented from zero in pure Rust.	25	Experimental	4	Rust
35	NextStat/nextstat.io Universal differentiable statistical inference engine (Rust + GPU)....	25	Experimental	7	Rust
36	tzervas/unsloth-rs Memory-optimized GPU kernels for LLM fine-tuning in Rust (2-5x speedup,...	23	Experimental	—	Rust
37	elftausend/gradients Deep Learning library written in Rust (OpenCL, CUDA & CPU)	22	Experimental	7	Rust
38	xpressgarageplus/RusTorch Build and run deep learning models in Rust with a PyTorch-like API that...	22	Experimental	—	Rust
39	torchforge-rs/torchforge-viz Lightweight, well-maintained training visualizer for Rust ML pipelines.	22	Experimental	—	Rust
40	sablokrep/datathon rust crate for optimized deep learning	22	Experimental	—	Rust
41	sablokrep/graphsage rust graphsage	21	Experimental	—	Rust
42	MDResearch/Mady 🌲 Ahead-of-time Static Macro-gen Automatic Differentiation. A little bit...	21	Experimental	9	Rust
43	springyworks/candlekos Fork focusing on new tensor ops for Candle: FFT and Scan, plus an...	21	Experimental	5	Rust
44	mattdeeds/irontensor Minimal Rust tensor library for training LLMs on Apple Silicon with Metal...	21	Experimental	—	Rust
45	arclabs561/propago Graph neural network primitives and training loops.	21	Experimental	—	Rust
46	arclabs561/structops Differentiable dynamic programming operators (Soft-DTW, soft shortest path)...	21	Experimental	—	Rust
47	lightsofapollo/safetensors-streaming Drop-in Rust replacement for safetensors that loads model weights faster	21	Experimental	—	Rust
48	arjunr00/ml-rust-cuda A set of GPU-accelerated machine learning algorithms and functions written...	21	Experimental	1	Rust
49	wesleyscholl/ZipGraph ⚡ Blazing-fast graph processing engine in Rust with ML-powered optimization....	20	Experimental	1	Rust
50	asukaminato0721/d2l.ai-rs use candle to implement some of the d2l.ai	20	Experimental	2	Rust
51	NickLucche/autograd.rs Simple Deep Learning library in Rust based on ndarray.	19	Experimental	4	Rust
52	wilhelmagren/tensorust Tensor library in Rust.	19	Experimental	4	Rust
53	vilhelmbergsoe/nn (WIP) Simple Deep Learning Framework and Auto Differentiation Engine in Rust	18	Experimental	2	Rust
54	matiasbattocchia/simple-grad Automatic differentiation in Rust for educational purposes. Autograd /...	18	Experimental	2	Rust
55	saivishwak/flowtorch Safe and Performant Scientific Computing Library	17	Experimental	1	Rust
56	BhavyeMathur/redstone-ml High-performance Machine Learning, Dynamic Auto-Differentiation and Tensor...	16	Experimental	17	Rust
57	james0248/tangent An autodiff engine in Rust with JAX-style tracing, forward/reverse-mode...	14	Experimental	1	Rust
58	samsja/rusty-grad Pytorch like framework in rust: Autograd with dynamic graphs for deep learning	13	Experimental	9	Rust
59	Selithrarion/ai-engine-deep-dive deconstructing llms, transformers, autograd, oonx, gguf, gpu kernels, cubecl	13	Experimental	—	Rust
60	dabevlohn/bfv_tensor_crypto Rust library with the Brakerski-Fan-Vercauteren (BFV) homomorphic encryption...	12	Experimental	1	Rust
61	bokutotu/curs cuda&cublas&cudnn wrapper for Rust	11	Experimental	4	Rust
62	Wiktor2718/matrix_flow Matrix Flow is a simple machine learning library written in Rust and CUDA....	11	Experimental	—	Rust
63	TheKindlerofWildfires/nonphysical Accelerated rust for scientific computing	11	Experimental	—	Rust
64	gursi26/autograd-rs An autograd implementation in Rust	11	Experimental	—	Rust
65	golanor/rust_graphs A rust library for analyzing graphs	11	Experimental	4	Rust