This is also related - https://arxiv.org/pdf/1909.01214.pdf (Mentioned in the paper of OP)