GitHub repository: HannesStark/3DInfomax
Primary language: Python
Description: Making self-supervised learning work on molecules by using their 3D geometry to pre-train GNNs. Implemented in DGL and Pytorch Geometric.