HannesStark / 3DInfomax

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

Related projects

Alternatives and complementary repositories for 3DInfomax