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

nanogemm

Minimalist, bare-metal SIMD & Assembly GEMM engine for Python. Sub-microsecond CPU matrix multiplication & Native MCP Server (nanogemm-mcp).

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.

ozaki-jax

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

photonic-mzi

Circuit-level simulator and bilingual visual explainer for validating photonic matrix multiply-accumulate with MZI meshes.

structlinks

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

pyclblast

Tuned OpenCL BLAS

matrox

Linear Algebra Library

sparsecootensormulpytorch

Pytorch Sparse COO Tensor Multiplication

pytrix

Python matrix DAO implemented in C

tenncor

Modular tensor algebra library