GitHub - cai4cai/torchsparsegradutils: A collection of...
A collection of utility functions to work with PyTorch sparse tensors. This is work-in-progress, here be dragons. Currenly available features with backprop include: Memory efficient sparse mm with batch support