Torch.jit.load