A PyTorch implementation of "A simple neural network module for relational reasoning" (Relational Networks)