The PyTorch way