torchjd
PublicLibrary for Jacobian descent with PyTorch. It enables optimization of neural networks with multiple losses (e.g. multi-task learning).
deep-learningjacobian-descentmulti-objective-optimizationmulti-task-learningmultiobjective-optimizationmultitask-learningoptimizationpythonpytorchtorch
Creat:2024-06-01T23:07:10
Update:2025-03-27T06:19:44
https://torchjd.org
261
Stars
1
Stars Increase