You may enjoy this paper: https://arxiv.org/abs/2307.14619