Library for Jacobian descent with PyTorch. It enables the optimization of neural networks with multiple losses (e.g. multi-task learning).
SimplexLab/TorchJD has 391 GitHub stars.
It currently ranks #31713 daily, #32084 weekly, #33607 monthly by momentum on Trending Repos; its best-ever rank is #24223.