Installation

TorchJD can be installed directly with pip:

pip install torchjd

Note that torchjd requires Python 3.10, 3.11, 3.12 or 3.13 and torch>=2.0.

Some aggregators (CAGrad and Nash-MTL) have additional dependencies that are not included by default when installing torchjd. To install them, you can use:

pip install torchjd[cagrad]

pip install torchjd[nash_mtl]

To install torchjd with all of its optional dependencies, you can also use:

pip install torchjd[full]