Sparse Tensors Python Packages

openequivariance

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

mlx-lattice

Sparse convolution library for MLX designed for point cloud operation.

minkowskiengine

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

openequivariance-extjax

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

spartan2

collection of data mining algorithms on big graphs and time series

sparsecootensormulpytorch

Pytorch Sparse COO Tensor Multiplication