Skip to content

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

License

Notifications You must be signed in to change notification settings

TorchJD/torchjd

Error
Looks like something went wrong!