https://arxiv.org/abs/1903.03894 https://arxiv.org/abs/2010.00577 PyG already has a support for this
https://arxiv.org/abs/1903.03894
https://arxiv.org/abs/2010.00577
PyG already has a support for this