AlphaBetaGamma96:

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


The engine expects to only compute gradients. So you do have to do two backwards right now.
You can write these formulas for sure but it might be tricky to be re-use the current engine for this