An implementation of the ideas from this paper https://arxiv.org/pdf/1803.10122.pdf - View it on GitHub