Hi, torch.autograd.functional.jacobian seems to be on the right path. Anyone knows if there is a counterpart when its first argument func is from a computation graph (like the forward function of a network, or the gradient)?