Tensor Python Packages

torch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

einops

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

simsimd

SIMD-accelerated distances, dot products, matrix ops, geospatial & geometric kernels for 16 numeric types — from 6-bit floats to 64-bit complex — across x86, Arm, RISC-V, and WASM, with bindings for Python, Rust, C, C++, Swift, JS, and Go 📐

cupy-cuda12x

NumPy & SciPy for GPU

autoray

Abstract your array operations.

cupy-cuda13x

NumPy & SciPy for GPU

numkong

SIMD-accelerated distances, dot products, matrix ops, geospatial & geometric kernels for 16 numeric types — from 6-bit floats to 64-bit complex — across x86, Arm, RISC-V, and WASM, with bindings for Python, Rust, C, C++, Swift, JS, and Go 📐

pytorch-tokenizers

On-device AI across mobile, embedded and edge for PyTorch

executorch

On-device AI across mobile, embedded and edge for PyTorch

cotengra

Hyper optimized contraction trees for large tensor networks and einsums

cupy-cuda11x

NumPy & SciPy for GPU

tensorly

TensorLy: Tensor Learning in Python.

tenseal

A library for doing homomorphic encryption operations on tensors

cupy

NumPy & SciPy for GPU

tensorly-torch

TensorLy-Torch: Deep Tensor Learning with TensorLy and PyTorch

cloud-volume

Read and write Neuroglancer datasets programmatically.

apache-tvm

Open Machine Learning Compiler Framework

tensorpac

Phase-Amplitude Coupling under Python

pymars

Mars is a tensor-based unified framework for large-scale data computation which scales numpy, pandas, scikit-learn and Python functions.

cupy-rocm-7-0

NumPy & SciPy for GPU

multidim-indexing

Batch multidimensional indexing for pytorch

rainbow-tensor

Colourful tensor operation visuals for NumPy style arrays, built for notebooks, docs, and teaching shape transformations.

pytat

Tensor Network State Packages

cupy-cuda102

NumPy & SciPy for GPU