TorchJD

TorchJD

🏢 Organization

1 repository on SrcLog

View on GitHub
1 Repos
270 Stars
10 Forks
270 Watchers

Repositories (1)

torchjd TorchJD/torchjd Python

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

270