You need sparse x sparse -> sparse multiplication, right? Right now we only have sparse x dense -> dense and sparse x dense -> sparse, because that’s what we needed for sparse Embedding updates. You can open a feature request if you want.