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.