TorchJD / torchjd

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

Related projects

Alternatives and complementary repositories for torchjd