Matrix Multiplication Python Packages

blis

💥 Fast matrix-multiplication as a self-contained Python library – no system dependencies!

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 📐

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 📐

sparse-dot-mkl

Python wrapper for Intel Math Kernel Library (MKL) matrix multiplication

ozaki-jax

FP64-accurate linear algebra on BF16/FP32 hardware (TPU) via the Ozaki scheme, in JAX.

trnblas

BLAS Levels 1–3 for AWS Trainium via NKI (cuBLAS-equivalent) — GEMM with stationary-tile reuse, batched GEMM, TRSM, validated DF-MP2 for quantum chemistry.

structlinks

Easily Access and visualize different Data structures including Linked lists, Doubly Linked lists, Binary trees, Graphs, Stacks, Queues, and Matrices.

pyclblast

Python bindings for CLBlast, the tuned OpenCL BLAS library

sparsecootensormulpytorch

Pytorch based sparse coo tensor multiplication (returns same elements with torch.matmul function but returns torch.tensor)

levi-edge

Drop-in small-matrix acceleration for PyTorch on edge devices

matrox

Linear Algebra library.

pytrix

Python matrix DAO implemented in C