1 result found Sort:

TorchJD
torchjd TorchJD
4
222
mit
4
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
251 commits to main branch, last one 6 hours ago