InstallationΒΆ

TorchJD can be installed directly with pip:

pip install torchjd

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