Sparse banded diagonal matrices for pytorch - View it on GitHub