| CARVIEW |
Select Language
HTTP/2 200
server: GitHub.com
content-type: text/html; charset=utf-8
last-modified: Sun, 16 Jul 2023 20:51:02 GMT
access-control-allow-origin: *
etag: W/"64b45836-2877"
expires: Sun, 28 Dec 2025 23:18:12 GMT
cache-control: max-age=600
content-encoding: gzip
x-proxy-cache: MISS
x-github-request-id: F81B:2F7ECD:8124AB:90FC13:6951B85C
accept-ranges: bytes
age: 0
date: Sun, 28 Dec 2025 23:08:12 GMT
via: 1.1 varnish
x-served-by: cache-bom-vanm7210054-BOM
x-cache: MISS
x-cache-hits: 0
x-timer: S1766963292.449581,VS0,VE220
vary: Accept-Encoding
x-fastly-request-id: 58a606304217d1c7d3a16c3840fc8e9fcef0306f
content-length: 2513
TorchDrug | A powerful and flexible machine learning platform for drug discovery
A powerful and flexible machine learning platform for drug discovery
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.
$ pip install torchdrug
Key Features
Minimal Domain Knowledge
Build and train machine learning models for drug discovery with minimal domain knowledge.
Datasets and Building Blocks
Empower fast iteration of ideas by a large collection of common datasets and building blocks.
Comprehensive Benchmarks
Benchmarks provide a systematic comparison of deep learning architectures for drug discovery.
Scalable Training and Inference
Seamlessly scale models to multiple CPUs, multiple GPUs, or even distributed settings.
TorchDrug
- © 2023 MilaGraph





