GitHub

MiDi: Mixed Graph and 3D Denoising Diffusion for Molecule Generation

This code was tested with PyTorch 2.0.1, cuda 11.8 and torch_geometric 2.3.1 on multiple gpus.