Alright, well I guess both of those are analogous for my purposes.
But… are the original inputs required to do autograd.grad ? Is it not possible to just call is in a similar way to backward where you just pass the loss and not the inputs that lead to it ?