@albanD thank your for your answer:) Can you point me to further information/elaborate a bit more how the PyTorch/Autograd internals work (conceptually) with regard to building the computation graph from the python code?