This is also relevant to your comment: https://arxiv.org/abs/1705.07663