← all repositories

DeepGraphLearning/torchdrug

A PyTorch-based machine learning library for graph neural networks applied to drug discovery and molecular property prediction.

1.6k stars Python Domain AppsML Frameworks
torchdrug
Velocity · 7d
+0.9
★ / day
Trend
steady
star history

TorchDrug is a PyTorch-based machine learning toolbox designed for drug discovery applications. It provides easy implementation of graph operations with GPU support and enables rapid prototyping of machine learning research using graph neural networks. The library is built on PyTorch and targets practitioners working with molecular data and protein structures.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.