Tensors Python Packages

quimb

A python library for quantum information and many-body calculations including tensor networks.

torchtyping

Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.

aesara

Aesara is a Python library for defining, optimizing, and efficiently evaluating mathematical expressions involving multi-dimensional arrays.

aesara-nightly

Aesara is a Python library for defining, optimizing, and efficiently evaluating mathematical expressions involving multi-dimensional arrays.

tntorch

Tensor Network Learning with PyTorch

varipeps

variPEPS -- Versatile tensor network library for variational ground state simulations in two spatial dimensions

nelux

NeLux: High-performance video processing for Python, powered by FFmpeg and PyTorch. Ultra-fast video decoding directly to tensors.

heat

Distributed tensors and Machine Learning framework with GPU and MPI acceleration in Python

braandket

BraAndKet is a library for numeral calculations of discrete quantum systems.

dimtensor

Unit-aware tensors for physics and scientific machine learning

celux

Lightspeed video decoding directly into tensors!

tensorshare

🤝 Trade any tensors over the network

torchstruct

Combine PyTorch tensors into a single structure.

pyttb

Python Tensor Toolbox

celux-cuda

HW accelerated video reading for ML Inference (CUDA version).

symmray

A minimal block sparse symmetric and fermionic tensor python library

torch-nested

Easily manipulate torch.Tensors inside highly nested data-structures.

forgenn

forgeNN is a neural network framework combining an autograd engine with a Keras‑like API. Developed by a college student.

okrolearn

A machine learning library created by Okerew, which is supposed to be a raw implementation of combining pytorch with scikit-learn

cowrie-py

Cowrie - Structured JSON binary codec with optional C extension (10x faster)

po-core-flyingpig

AI system integrating philosophers as dynamic tensors for responsible meaning generation

ttpy

Python implementation of the TT-Toolbox

torchsaber

Elegant dimensions for a more civilized age

whichaxis

NumPy, but you don’t forget which axis is which.