PyTorch 2.1: Quansight’s Improvements to BSR Sparse Matrix Multiplication

This post presents Quansight work implementing new block sparse row (BSR) kernels for sparse matrix multiplication in PyTorch 2.1.
This post presents Quansight work implementing new block sparse row (BSR) kernels for sparse matrix multiplication in PyTorch 2.1.