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 ...
Library for Jacobian descent with PyTorch. It enables optimization of neural networks with multiple losses (e.g. multi-task learning).