TensorFlow implementation: https://github.com/jimfleming/recurrent-entity-networks