TorchDrug

Use TorchDrug as a modular PyTorch graph-learning stack:

1. load a datasets.* dataset,
2. choose a models.* representation model,
3. wrap it in a tasks.* objective,
4. train and evaluate it with core.Engine.

The current official documentation and latest published release are both 0.2.1 (released July 2023; rechecked October 1, 2026). Treat newer Python or PyTorch combinations as unverified rather than silently assuming compatibility.

Start with the version guard

Before generating or debugging code, inspect the environment: