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 64 tensor algebra frameworks tracked. 4 score above 70 (verified tier). The highest-rated is pymc-devs/pytensor at 87/100 with 598 stars and 898,647 monthly downloads. 1 of the top 10 are actively maintained.

Get all 64 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
1 pymc-devs/pytensor

PyTensor allows you to define, optimize, and efficiently evaluate...

87
Verified
2 arogozhnikov/einops

Flexible and powerful tensor operations for readable and reliable code (for...

77
Verified
3 tensorpack/tensorpack

A Neural Net Training Interface on TensorFlow, with focus on speed + flexibility

72
Verified
4 tensorly/tensorly

TensorLy: Tensor Learning in Python.

70
Verified
5 tensorlayer/TensorLayer

Deep Learning and Reinforcement Learning Library for Scientists and Engineers

69
Established
6 lava-nc/lava-dl

Deep Learning library for Lava

64
Established
7 tensorlayer/TensorLayerX

TensorLayerX: A Unified Deep Learning and Reinforcement Learning Framework...

60
Established
8 google-deepmind/sonnet

TensorFlow-based neural network library

58
Established
9 lava-nc/lava

A Software Framework for Neuromorphic Computing

57
Established
10 taehoonlee/tensornets

High level network definitions with pre-trained weights in TensorFlow

52
Established
11 NVIDIA/framework-reproducibility

Providing reproducibility in deep learning frameworks

52
Established
12 BlackSamorez/tensor_parallel

Automatically split your PyTorch models on multiple GPUs for training & inference

51
Established
13 tensorflow/tensor2tensor

Library of deep learning models and datasets designed to make deep learning...

51
Established
14 numenta/nupic-legacy

Numenta Platform for Intelligent Computing is an implementation of...

51
Established
15 zwang4/awesome-machine-learning-in-compilers

Must read research papers and links to tools and datasets that are related...

49
Emerging
16 pyfenn/fenn

Friendly Environment for Neural Networks (fenn) is a simple framework that...

49
Emerging
17 parrt/tensor-sensor

The goal of this library is to generate more helpful exception messages for...

47
Emerging
18 joserapa98/tensorkrowch

Smooth integration of tensor networks in machine learning

46
Emerging
19 neuroptimiser/neuroptimiser

NeurOptimiser is a neuromorphic optimisation framework in which...

46
Emerging
20 isxcode/torch-yun

Ultra-Intelligent AI-in-one Application Platform | 至数云-超轻量级人工智能应用平台/智能平台

43
Emerging
21 tensorpool/tensorpool

The CLI for GPUs

40
Emerging
22 marcoloco23/dimtensor

Unit-aware tensors for physics and scientific machine learning

40
Emerging
23 tariqdaouda/Mariana

The Cutest Deep Learning Framework which is also a wonderful Declarative Language

39
Emerging
24 lanl/T-ELF

Tensor Extraction of Latent Features (T-ELF). Within T-ELF's arsenal are...

38
Emerging
25 open-neuromorphic/open-neuromorphic

A list of neuromorphic software projects

38
Emerging
26 david-leon/Dandelion

A light weight deep learning framework, on top of Theano, offering better...

38
Emerging
27 derinworks/penr-oz-neural-network-v3-torch-ddp

Version 3 Implementation of Neural Network service leveraging pytorch...

38
Emerging
28 monkbai/DNN-decompiler

BTD - Bin To DNN: A DNN Executables Decompiler

38
Emerging
29 howl-anderson/tensorweaver

A modern educational deep learning framework for students, engineers and researchers

37
Emerging
30 uclnlp/ntp

End-to-End Differentiable Proving

37
Emerging
31 opendilab/DI-treetensor

Let DI-treetensor help you simplify the structure...

35
Emerging
32 TheRemyyy/neurox-ai

Neuromorphic Computing System GPU-accelerated, spiking neural network...

34
Emerging
33 yhtang/FunFact

Tensor decomposition with arbitrary expressions: inner, outer, elementwise...

34
Emerging
34 ArjunSahlot/TensorNN

Python machine learning library made from scratch.

33
Emerging
35 Ch0ser/Deepflows

DeepFlows is a lightweight deep learning framework for teaching and...

32
Emerging
36 Dristro/puretorch

Raw implementation of PyTorch using NumPy. Structured similar to PyTorch,...

32
Emerging
37 tensorlang/tensorlang

Tensorlang, a differentiable programming language based on TensorFlow

32
Emerging
38 wagamamaz/tensorlayer-tricks

How to use TensorLayer

32
Emerging
39 csem/NeurIO

Open source Python library for deploying deep learning model on Edge devices

32
Emerging
40 chainyo/tensorshare

🤝 Trade any tensors over the network

31
Emerging
41 RJT1990/mantra

A high-level, rapid development framework for machine learning projects

31
Emerging
42 hpcaitech/TensorNVMe

A Python library transfers PyTorch tensors between CPU and NVMe

31
Emerging
43 Yehor-Mishchyriak/PureML

Transparent, NumPy-only deep learning framework for teaching, small-scale...

31
Emerging
44 graphcore/poptorch

PyTorch interface for the IPU

31
Emerging
45 danielsabinasz/TensorSlow

Re-implementation of TensorFlow in pure python, with an emphasis on code...

30
Emerging
46 tongnie/tensorlib

Repository for paper 'Truncated tensor Schatten p-norm based approach for...

30
Emerging
47 etrommer/torch-approx

GPU-accelerated Neural Network layers using Approximate Multiplications for PyTorch

29
Experimental
48 by321/safetensors_util

Utility for Safetensors Files

28
Experimental
49 TheLeprechaun25/NCOLib

The Neural Combinatorial Optimization Library (NCOLib) is an accessible...

26
Experimental
50 oortega20/tensorslow

Machine Learning Library using Python Lists

24
Experimental
51 attentionmech/tensorlens

aesthetic tensor visualiser

22
Experimental
52 rageelixer853/torchcomms

📡 Streamline PyTorch communications with torchcomms, an experimental API...

22
Experimental
53 Danik277/dimtensor

🔧 Enable safe computations with unit-aware tensors for physics and machine...

22
Experimental
54 rvermillion/tensile

Dual-backend ML framework for MLX and PyTorch

22
Experimental
55 WillD-cell/spider-compute

Universal compute layer for next-gen AI chips

22
Experimental
56 masc-it/gradientlab

gradientlab is a clean space to do deep learning experiments

21
Experimental
57 AllenNeuralDynamics/aind-torch-utils

Generic, queue-based, multi-GPU PyTorch inference pipeline for large...

20
Experimental
58 yuya-isaka/HPU

Accelerator for Hyperdimensional Computing (HDC)

19
Experimental
59 copev313/TensorTonic-Solutions

A collection of solutions to TensorTonic problems.

15
Experimental
60 graphcore-research/tessellate-ipu

TessellateIPU: low level Poplar tile programming from Python

14
Experimental
61 tarun-rai21/TensorTonic-Solutions

My solutions to TensorTonic problems. TensorTonic is an interactive,...

14
Experimental
62 thesecretrapper786-netizen/catalyst-n1

Implement a configurable neuromorphic processor with LIF neurons, STDP...

14
Experimental
63 derinworks/penr-oz-neural-network-torch

Implementation of Neural Network service with pytorch library

11
Experimental
64 Frikallo/x-ops

Simple yet powerful tensor manipulation library with a clear and verbose...

10
Experimental