Sparse Convolution Python Packages

minkowskiengine

Minkowski Engine is an auto-diff neural network library for high-dimensional sparse tensors

sparsepixels

Efficient convolution for sparse data on FPGAs

spconv-triton

spconv-Triton is a fast sparse convolution library with full operator support (SubM conv, Conv3D, Transposed Conv, Pooling). It is hardware-agnostic and runs on GPUs supporting Triton.

sige

[NeurIPS 2022, T-PAMI 2023] Efficient Spatially Sparse Inference for Conditional GANs and Diffusion Models

sparselayer-tensorflow

An implementation of Sparse Layers in TensorFlow 2. x.