Linear Algebra Python Packages

blis

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

xarray-einstats

Stats, linear algebra and einops for xarray

qdldl

Python interface to the QDLDL (https://github.com/osqp/qdldl) free LDL factorization routine for quasi-definite linear systems

transformations

Homogeneous transformation matrices and quaternions.

lineax

Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax

pylibraft-cu12

RAFT contains fundamental widely-used algorithms and primitives for machine learning and information retrieval. The algorithms are CUDA-accelerated and form building blocks for more easily writing high performance applications.

libraft-cu12

RAFT contains fundamental widely-used algorithms and primitives for machine learning and information retrieval. The algorithms are CUDA-accelerated and form building blocks for more easily writing high performance applications.

raft-dask-cu12

RAFT contains fundamental widely-used algorithms and primitives for machine learning and information retrieval. The algorithms are CUDA-accelerated and form building blocks for more easily writing high performance applications.

scikit-spatial

Spatial objects and computations based on NumPy arrays.

discretize

Discretization tools for finite volume and inverse problems.

pymatsolver

Solve matrix equations in python.

libraft-cu13

RAFT contains fundamental widely-used algorithms and primitives for machine learning and information retrieval. The algorithms are CUDA-accelerated and form building blocks for more easily writing high performance applications.

pylibraft-cu13

RAFT contains fundamental widely-used algorithms and primitives for machine learning and information retrieval. The algorithms are CUDA-accelerated and form building blocks for more easily writing high performance applications.

vg

Vector-geometry toolbelt for 3D points and vectors

python-graphblas

Python library for GraphBLAS: high-performance sparse linear algebra for scalable graph analytics

pentapy

A Python toolbox for pentadiagonal linear systems

gf2-lin-algebra

Python package for linear algebra operations in GF(2)

simpeg

Simulation and Parameter Estimation in Geophysics - A python package for simulation and gradient based parameter estimation in the context of geophysical applications.

rmath-py

Drop-in numerical accelerator for Python, powered by Rust. High-performance arrays, linear algebra, and statistics — outside the GIL.

gdmath

🚀Spatium: a fast python spatial math library

openmtpk

Hardware Accelerated General Purpose Mathematics Package (very stale)

imate

Implicit Matrix Trace Estimator. Computes the trace of functions of implicit matrices. Accelerated on CPU and CUDA-capable GPU devices.

pycapacity

An efficient task-space capacity calculation package for robotics and biomechanics

trnsparse

Sparse matrix operations for AWS Trainium via NKI (cuSPARSE-equivalent) — CSR/COO formats, SpMV and SpMM via gather-matmul-scatter, Schwarz integral screening for quantum chemistry.