torchjd
A PyTorch library for Jacobian descent that supports optimization of neural networks with multiple losses.
Contributor to torchjd, a PyTorch library for Jacobian descent and multi-loss neural network optimization.
- Role: Contributor
- Language: Python
- Repository: torchjd
- Homepage: torchjd.org