Search 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

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 based sparse coo tensor multiplication (returns same elements with torch.matmul function but returns torch.tensor)