/tools/3DInfomax
3DInfomax
HannesStark/3DInfomax
172 stars33 forksPythonAdded February 5, 2026
summary
3DInfomax enhances graph neural networks for predicting molecular properties by leveraging 3D molecular geometry. It provides tools for pre-training and fine-tuning models on various molecular datasets, enabling better predictions and molecular fingerprint generation.
description
Making self-supervised learning work on molecules by using their 3D geometry to pre-train GNNs. Implemented in DGL and Pytorch Geometric.
topics
graph-representation-learningself-supervised-learninggraphmolecules3dpytorch-geometricdgl-graphgnngraph-neural-networks
Ratings
N/A
0 ratings
Rate this tool: