TorchDrug: A Powerful and Flexible Machine Learning Platform for Drug Discovery

Follow the full discussion on Reddit.
TorchDrug is a machine learning platform designed for drug discovery, covering techniques from graph machine learning (graph neural networks, geometric deep learning & knowledge graphs), deep generative models to reinforcement learning. It provides a comprehensive and flexible interface to support rapid prototyping of drug discovery models in PyTorch.