Maybe someone finds this interesting. This paper seems to be related: https://arxiv.org/abs/1907.04490