AIbase
Product LibraryTool Navigation

torchjd

Public

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

Creat2024-06-01T23:07:10
Update2025-03-27T06:19:44
https://torchjd.org
222
Stars
0
Stars Increase