pytorch-block-sparse

PyTorch extension for fast block sparse matrices computation, drop in replacement for torch.nn.Linear.


Keywords
PyTorch, sparse, matrices, machine, learning
Licenses
BSD-3-Clause/BSD-3-Clause
Install
pip install pytorch-block-sparse==0.1.2