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