Sparse Matrix Python Packages

sparse

Sparse multi-dimensional arrays for the PyData ecosystem

sparse-dot-topn

Python package to accelerate the sparse matrix multiplication and top-n similarity selection

pyamg

Algebraic Multigrid Solvers in Python

python-graphblas

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

fast-matrix-market

Fast and full-featured Matrix Market I/O library for C++, Python, and R

csr

Compressed sparse matrices

brainevent

Event-driven Computation for Brain Dynamics.

sparse-dot-mkl

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

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.

grblas

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

spector

Sparse vectors.

matrepr

Format matrices and tensors to HTML, string, and LaTeX, with Jupyter integration.

torchsparsegradutils

A collection of utility functions to work with PyTorch sparse tensors

torch-sla

torch sparse linear algebra library

pyrsb

PyRSB: a Python interface to the librsb Sparse Matrix library

annnet

A Python tool to build, manipulate, and interoperate with flexible graph structures

sansa

SANSA - sparse EASE for millions of items

coolpuppy

A versatile tool to perform pile-up analysis on Hi-C data in .cool format.

algebrax

Algebraic Structures in pure Python

dhl-mm

Up to 913x faster Lie algebra products for all five exceptional algebras with differentiable PyTorch equivariant layers

indigo

High-Performance Structured Linear Operators

betterspy

:eyes: Visualization for SciPy sparse matrices.

pycubool

Sparse linear Boolean algebra for Nvidia Cuda

matspy

Sparse matrix spy plot and sparkline renderer that works with Jupyter.