1 result found Sort:

TorchJD
torchjd TorchJD
1
167
mit
5
Library for Jacobian descent with PyTorch. It enables optimization of neural networks with multiple losses (e.g. multi-task learning).
torch python pytorch optimization deep-learning jacobian-descent multitask-learning multi-task-learning multiobjective-optimization multi-objective-optimization
Created 2024-06-01
185 commits to main branch, last one 7 days ago