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