A colab notebook re-implementing the Neural Relational Inference method. - View it on GitHub