How to trace torch.autograd.backward or torch.autograd.grad?

[shuokay]

These discussions might provide some help: How does torch.compile work with autograd? - #4 by Chillee