Graph networks as learnable physics engines for inference and control
A simple neural network module for relational reasoning