autojacΒΆ

This package enables Jacobian descent, through the backward and mtl_backward functions, which are meant to replace the call to torch.backward or loss.backward in gradient descent. To combine the information of the Jacobian, an aggregator from the aggregation package has to be used.