Tensor Algebra Frameworks

Libraries and frameworks for tensor operations, decomposition, manipulation, and mathematical expression optimization across multiple backends (NumPy, PyTorch, JAX, TensorFlow). Does NOT include general deep learning frameworks, neural network architectures, or domain-specific applications like computer vision or NLP models.

There are 68 tensor algebra frameworks tracked. 2 score above 70 (verified tier). The highest-rated is pymc-devs/pytensor at 83/100 with 598 stars. 2 of the top 10 are actively maintained.

Get all 68 projects as JSON

curl "https://pt-edge.onrender.com/api/v1/datasets/quality?domain=ml-frameworks&subcategory=tensor-algebra-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	pymc-devs/pytensor PyTensor allows you to define, optimize, and efficiently evaluate...	83	Verified	598	Python
2	arogozhnikov/einops Flexible and powerful tensor operations for readable and reliable code (for...	70	Verified	9,425	Python
3	lava-nc/lava-dl Deep Learning library for Lava	68	Established	174	Jupyter Notebook
4	tensorly/tensorly TensorLy: Tensor Learning in Python.	67	Established	1,669	Python
5	tensorpack/tensorpack A Neural Net Training Interface on TensorFlow, with focus on speed + flexibility	62	Established	6,295	Python
6	tensorlayer/TensorLayerX TensorLayerX: A Unified Deep Learning and Reinforcement Learning Framework...	62	Established	527	Python
7	tensorlayer/TensorLayer Deep Learning and Reinforcement Learning Library for Scientists and Engineers	61	Established	7,393	Python
8	lava-nc/lava A Software Framework for Neuromorphic Computing	61	Established	695	Jupyter Notebook
9	taehoonlee/tensornets High level network definitions with pre-trained weights in TensorFlow	59	Established	1,000	Python
10	google-deepmind/sonnet TensorFlow-based neural network library	58	Established	9,907	Python
11	zwang4/awesome-machine-learning-in-compilers Must read research papers and links to tools and datasets that are related...	56	Established	1,659	—
12	tensorflow/tensor2tensor Library of deep learning models and datasets designed to make deep learning...	51	Established	17,062	Python
13	numenta/nupic-legacy Numenta Platform for Intelligent Computing is an implementation of...	51	Established	6,357	Python
14	BlackSamorez/tensor_parallel Automatically split your PyTorch models on multiple GPUs for training & inference	50	Established	656	Python
15	pyfenn/fenn Friendly Environment for Neural Networks (fenn) is a simple framework that...	50	Established	39	Python
16	isxcode/torch-yun Ultra-Intelligent AI-in-one Application Platform \| 至数云-超轻量级人工智能应用平台/智能平台	50	Established	24	Java
17	NVIDIA/framework-reproducibility Providing reproducibility in deep learning frameworks	50	Established	434	Python
18	joserapa98/tensorkrowch Smooth integration of tensor networks in machine learning	48	Emerging	49	Python
19	neuroptimiser/neuroptimiser NeurOptimiser is a neuromorphic optimisation framework in which...	48	Emerging	2	Python
20	parrt/tensor-sensor The goal of this library is to generate more helpful exception messages for...	48	Emerging	810	Jupyter Notebook
21	tariqdaouda/Mariana The Cutest Deep Learning Framework which is also a wonderful Declarative Language	46	Emerging	150	JavaScript
22	open-neuromorphic/open-neuromorphic A list of neuromorphic software projects	45	Emerging	287	—
23	tensorpool/tensorpool The CLI for GPUs	44	Emerging	171	Python
24	uclnlp/ntp End-to-End Differentiable Proving	44	Emerging	90	NewLisp
25	howl-anderson/tensorweaver A modern educational deep learning framework for students, engineers and researchers	44	Emerging	7	Jupyter Notebook
26	marcoloco23/dimtensor Unit-aware tensors for physics and scientific machine learning	44	Emerging	3	Python
27	monkbai/DNN-decompiler BTD - Bin To DNN: A DNN Executables Decompiler	42	Emerging	200	Python
28	derinworks/penr-oz-neural-network-v3-torch-ddp Version 3 Implementation of Neural Network service leveraging pytorch...	42	Emerging	1	Python
29	opendilab/DI-treetensor Let DI-treetensor help you simplify the structure...	42	Emerging	213	Python
30	yhtang/FunFact Tensor decomposition with arbitrary expressions: inner, outer, elementwise...	41	Emerging	59	Python
31	lanl/T-ELF Tensor Extraction of Latent Features (T-ELF). Within T-ELF's arsenal are...	41	Emerging	25	Python
32	david-leon/Dandelion A light weight deep learning framework, on top of Theano, offering better...	40	Emerging	16	Python
33	tensorlang/tensorlang Tensorlang, a differentiable programming language based on TensorFlow	39	Emerging	292	Python
34	csem/NeurIO Open source Python library for deploying deep learning model on Edge devices	39	Emerging	1	C
35	Dristro/puretorch Raw implementation of PyTorch using NumPy. Structured similar to PyTorch,...	39	Emerging	8	Python
36	wagamamaz/tensorlayer-tricks How to use TensorLayer	39	Emerging	348	—
37	RJT1990/mantra A high-level, rapid development framework for machine learning projects	38	Emerging	344	Python
38	TheRemyyy/neurox-ai Neuromorphic Computing System GPU-accelerated, spiking neural network...	38	Emerging	4	Rust
39	hpcaitech/TensorNVMe A Python library transfers PyTorch tensors between CPU and NVMe	38	Emerging	125	C++
40	graphcore/poptorch PyTorch interface for the IPU	38	Emerging	180	C++
41	Yehor-Mishchyriak/PureML Transparent, NumPy-only deep learning framework for teaching, small-scale...	37	Emerging	10	Python
42	tongnie/tensorlib Repository for paper 'Truncated tensor Schatten p-norm based approach for...	37	Emerging	16	Python
43	danielsabinasz/TensorSlow Re-implementation of TensorFlow in pure python, with an emphasis on code...	37	Emerging	678	Jupyter Notebook
44	ArjunSahlot/TensorNN Python machine learning library made from scratch.	37	Emerging	2	Python
45	Ch0ser/Deepflows DeepFlows is a lightweight deep learning framework for teaching and...	36	Emerging	38	Python
46	etrommer/torch-approx GPU-accelerated Neural Network layers using Approximate Multiplications for PyTorch	36	Emerging	10	Python
47	by321/safetensors_util Utility for Safetensors Files	35	Emerging	100	Python
48	TheLeprechaun25/NCOLib The Neural Combinatorial Optimization Library (NCOLib) is an accessible...	33	Emerging	19	Python
49	chainyo/tensorshare 🤝 Trade any tensors over the network	32	Emerging	31	Python
50	attentionmech/tensorlens aesthetic tensor visualiser	29	Experimental	27	Jupyter Notebook
51	oortega20/tensorslow Machine Learning Library using Python Lists	27	Experimental	2	Python
52	yuya-isaka/HPU Accelerator for Hyperdimensional Computing (HDC)	26	Experimental	35	C
53	AllenNeuralDynamics/aind-torch-utils Generic, queue-based, multi-GPU PyTorch inference pipeline for large...	26	Experimental	1	Python
54	masc-it/gradientlab gradientlab is a clean space to do deep learning experiments	25	Experimental	2	Jupyter Notebook
55	rvermillion/tensile Dual-backend ML framework for MLX and PyTorch	22	Experimental	—	Python
56	WillD-cell/spider-compute Universal compute layer for next-gen AI chips	22	Experimental	—	Python
57	Danik277/dimtensor 🔧 Enable safe computations with unit-aware tensors for physics and machine...	22	Experimental	—	Python
58	rageelixer853/torchcomms 📡 Streamline PyTorch communications with torchcomms, an experimental API...	21	Experimental	—	C++
59	graphcore-research/tessellate-ipu TessellateIPU: low level Poplar tile programming from Python	21	Experimental	13	Python
60	Frikallo/x-ops Simple yet powerful tensor manipulation library with a clear and verbose...	17	Experimental	1	Python
61	copev313/TensorTonic-Solutions A collection of solutions to TensorTonic problems.	17	Experimental	1	Python
62	nebHailemariam/NebTorch NebTorch—a minimal Autograd engine built from scratch using NumPy	15	Experimental	9	Python
63	thesecretrapper786-netizen/catalyst-n1 Implement a configurable neuromorphic processor with LIF neurons, STDP...	14	Experimental	—	—
64	tarun-rai21/TensorTonic-Solutions My solutions to TensorTonic problems. TensorTonic is an interactive,...	14	Experimental	—	Python
65	nickovchinnikov/microtorch MicroTorch: A lightweight autograd library supporting both CPU and GPU...	14	Experimental	7	Jupyter Notebook
66	derinworks/penr-oz-neural-network-torch Implementation of Neural Network service with pytorch library	13	Experimental	—	Python
67	K-Wu/HET HET: The HET Hetero-GNN Kernel Optimization and Code Generation Project	12	Experimental	5	C++
68	NeuralFlogo/framework This framework streamlines the design, training, evaluation, and deployment...	11	Experimental	—	Python

Comparisons in this category

TensorLayerX and TensorLayer (62 vs 61) lava-dl and lava (68 vs 61)