PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations

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

SPLADE: sparse neural search (SIGIR21, SIGIR22)

A model compression and acceleration toolbox based on pytorch.

[NeurIPS 2022] Efficient Spatially Sparse Inference for Conditional GANs and Diffusion Models

Sparse and structured neural attention mechanisms

A cool place to store your Hi-C

SparseTIR: Sparse Tensor Compiler for Deep Learning

Attention-guided CNN for image denoising(Neural Networks,2020)

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 writ…

Efficient matrix representations for working with tabular data

The RAG Experiment Accelerator is a versatile tool designed to expedite and facilitate the process of conducting experiments and evaluations using Azure Cognitive Search and RAG pattern.

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

A sparse KLU solver for PyTorch.

Source code for "Sparse in Space and Time: Audio-visual Synchronisation with Trainable Selectors." (Spotlight at the BMVC 2022)

Efficient forward- and reverse-mode sparse Jacobians using Jax

Sparse learning in TensorFlow using data acquired from Spark.

Metrically Scaled Monocular Depth Estimation through Sparse Priors for Underwater Robots

bare bone Hierarchial Temporal Memory

JSparse is a high-performance auto-differentiation library for sparse voxels computation and point cloud processing based on TorchSparse and Jittor.