Is there a general way calculate the Jacobian of a module, in a way that we retain the graph and can backprop through operations on the Jacobian as well?
Something similar to how WGAN operates but defined on more than a single output.