We’re tracking the finishing of this feature here: https://github.com/pytorch/pytorch/issues/2389 628
For torch.mm in particular, it wasn’t that hard for me to make it work with autograd. you can probably patch these in while I send a PR with unit tests and all: https://gist.github.com/anonymous/49c10bc17ac4a97307d52c07d01a2870 279