blocksparse-pytorch-implement pytorch implements block sparse How to use Compile operator into python libs at the root directory, use python setup.py build_ext --inplace to compile (DON'T USE NINJA, i don't know why it failed when I use it) Try test_api.py it gives a example to use.