I was wondering if it were at all possible to implement a backward of backward method in an autograd function?

Hi, plain nn.Module don’t call backward by themselves.
You might need to update the library you’re using on top of pytorch to stop doing that.
Then you can use autograd.grad() to do what you want.