Relation Networks for CLEVR implemented in PyTorch - View it on GitHub