A PyG re-implementation of NBFNet can be found here. You may install the dependencies via either conda or pip. Generally, NBFNet works with Python 3.7/3.8 and PyTorch ...