Code for our paper on doing resource allocation with graph neural networks - View it on GitHub