/crystal-gnn-pyg

Primary LanguagePythonMIT LicenseMIT

🌐 Crystal Graph Neural Networks 💎

A banchmarking framework for Crystal Graph Neural Networks (CGNN)

built on top of PyTorch Lightning and Torch Geometric (PYG)

Under Construction 🚧