Code implementation of the paper Relational Deep Reinforcement Learning ( https://arxiv.org/abs/1806.01830 )