Add parameter `create_graph` to `Grad` and `Jac`

TorchJD / torchjd

Library for Jacobian descent with PyTorch. It enables optimization of neural networks with multiple losses (e.g. multi-task learning).

https://torchjd.org

MIT License

151 stars 0 forks source link

Closed PierreQuinton closed 2 months ago