TorchDrug: A Powerful and Flexible Machine Learning Platform for Drug Discovery
[article]
Zhaocheng Zhu, Chence Shi, Zuobai Zhang, Shengchao Liu, Minghao Xu, Xinyu Yuan, Yangtian Zhang, Junkun Chen, Huiyu Cai, Jiarui Lu, Chang Ma, Runcheng Liu
(+3 others)
2022
arXiv
pre-print
State-of-the-art techniques based on geometric deep learning (or graph machine learning), deep generative models, reinforcement learning and knowledge graph reasoning are implemented for these tasks. ...
TorchDrug benchmarks a variety of important tasks in drug discovery, including molecular property prediction, pretrained molecular representations, de novo molecular design and optimization, retrosynthsis ...
The most prominent packages are PyTorch-Geometric (PyG) (Fey and Lenssen, 2019) and Deep Graph Library (DGL) (Wang et al., 2019) , which are targeted at building graph neural networks (GNNs) in PyTorch ...
arXiv:2202.08320v1
fatcat:jzbsxbhizjgwflwvh2avoyzcai