pytorch-block-sparse (pypi): PyTorch extension for fast block sparse matrices computation, drop in replacement for torch.nn.Linear.
Keywords: PyTorch|sparse|matrices|machine learning