jacobian-descent topic
List
jacobian-descent repositories
torchjd
277
Stars
11
Forks
277
Watchers
Library for Jacobian descent with PyTorch. It enables the optimization of neural networks with multiple losses (e.g. multi-task learning).
