TorchDrug - A deep learning platform for drug discovery#

TorchDrug is a graph representation learning toolbox built on top of PyTorch platform. It provides a comprehensive and flexible interface to support rapid protoyping of deep learning models for graph structured data. TorchDrug covers latest models and benchmarks, as well as approachable and extensible building blocks.