Sparse Tensors Python Packages

openequivariance

OpenEquivariance: a fast, open-source GPU JIT kernel generator for the Clebsch-Gordon Tensor Product.

minkowskiengine

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

openequivariance-extjax

JAX C++ Extension for OpenEquivariance

spartan2

A collection of data mining algorithms on big graphs and time series

sparsecootensormulpytorch

Pytorch Sparse COO Tensor Multiplication