-
Notifications
You must be signed in to change notification settings - Fork 7
Library for Jacobian descent with PyTorch. It enables the optimization of neural networks with multiple losses (e.g. multi-task learning).
License
TorchJD/torchjd
ErrorLooks like something went wrong!
About
Library for Jacobian descent with PyTorch. It enables the optimization of neural networks with multiple losses (e.g. multi-task learning).