Hm, that computation should only take a fraction of a second, and I am not sure why you see all the verbose output when running y=x.mm(w1). Also, as far as I know, PyTorch doesn’t use NumPy for any of its computations but has it’s own tensor library (from torch, and aten) to do things like matrix multiplication.