TorchDrug
TorchDrug is a graph representation learning toolbox built on top of the PyTorch platform.
1543
212
TorchDrug provides a comprehensive and flexible interface to support the rapid prototyping of deep learning models for graph-structured data. TorchDrug covers the latest models and benchmarks, as well as approachable and extensible building blocks.
Official Website Link : https://torchdrug.ai/
Links
Documentation
Official Documentation and Resources
De Novo Molecule Generation with GCPNs using TorchDrug
A report showcasing how reinforcement learning, specifically graph convolutional policy networks, can help create brand new molecules to treat real-world diseases
Tutorials & Publications
A set of Tutorials and Publications related to TorchDrug
Iterate on AI agents and models faster. Try Weights & Biases today.