The key difference with current graph deep learning libraries, such as PyTorch Geometric (PyG) and Deep Graph Library (DGL), is that, while PyG and DGL support basic graph deep learning operations, DIG provides a unified testbed for higher level, research-oriented graph deep learning tasks, such as graph generation, self-supervised learning, explainability, 3D graphs, and graph out-of-distribution. If you are working or plan to work on research in graph deep learning, DIG enables you to develop your own methods within our extensible framework, and compare with current baseline methods using common datasets and evaluation metrics without extra efforts. It includes unified implementations of data interfaces, common algorithms, and evaluation metrics for several advanced tasks. Our goal is to enable researchers to easily implement and benchmark algorithms.
Features
- Graph Generation
- Self-supervised Learning on Graphs
- Explainability of Graph Neural Networks
- Deep Learning on 3D Graphs
- Graph OOD
- Run SphereNet on QM9 to incorporate 3D information of molecules
- Dive into Graphs is a turnkey library for graph deep learning research