/3DInfomax

Making self-supervised learning work on molecules by using their 3D geometry to pre-train GNNs. Implemented in DGL and Pytorch Geometric.

Primary LanguagePython

Stargazers

No one’s star this repository yet.