I think you will like this paper then: https://arxiv.org/abs/1711.09576