Awesome thanks.
Any chance we can also see where the interior nodes’ derivatives (those with grad_fn() == 1 are computed (the ones that we use to calculate these gradients to accumulate in leaves)? Is it inside libtorch.so?