See technical details in the paper: https://arxiv.org/abs/2209.01188