ahhh fair enough, that makes more sense!
But whether it’s jvp or vjp the actual formula within that method is just the derivative of the output of the module w.r.t its input? Is that correct? Or is there a difference when using forward-ad?
Thank you for the help!