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 release are both 0.2.1. 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: