You don’t need to do anything. As long as you are using PyTorch operations and don’t break the computation graph, Autograd will track all operations under the hood for you. There is no explicit call for it.