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.

libcuvs-cu12

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

torch-sparse

PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations

cuvs-cu12

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

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

genieclust

Genie: Fast and Robust Hierarchical Clustering

scikit-sundae

Python bindings to SUNDIALS differential algebraic equation solvers

sparse-dot-mkl

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

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

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.

speedtorch

Fast Pinned CPU -> GPU transfer

grblas

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

npdict

A Python dictionary wrapper for numpy arrays

fastrvm

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

matrepr

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

pylibraft-cu11

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-cu11

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.