Sparse Python Packages

tabmat

Efficient matrix representations for working with tabular data

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.

cuvs-cu12

cuVS - a library for vector search and clustering on the GPU

libcuvs-cu12

cuVS - a library for vector search and clustering on the GPU

torch-sparse

PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations

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.

python-graphblas

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

cooler

A cool place to store your Hi-C

scikit-sundae

Python bindings to SUNDIALS differential algebraic equation solvers

genieclust

Genie: Fast and Robust Hierarchical Clustering

sparse-dot-mkl

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

libcuvs-cu13

cuVS - a library for vector search and clustering on the GPU

mlx-lattice

Sparse convolution library for MLX designed for point cloud operation.

raft-dask-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.

tmu

Implements the Tsetlin Machine, Coalesced Tsetlin Machine, Convolutional Tsetlin Machine, Regression Tsetlin Machine, and Weighted Tsetlin Machine, with support for continuous features, drop clause, Type III Feedback, focused negative sampling, multi-task classifier, autoencoder, literal budget, and one-vs-one multi-class classifier. TMU is written in Python with wrappers for C and CUDA-based clause evaluation and updating.

libsps

O(1) region count queries using sparse prefix sums

npdict

Python Package for Dictionary Wrappers for Numpy Arrays

fastrvm

[fastrvm] Relevance Vector Machine in Python with a C++ core

cuvs-cu13

cuVS - a library for vector search and clustering on the GPU

grblas

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

speedtorch

Library for faster pinned CPU <-> GPU transfer in Pytorch