Have you seen this paper -> https://arxiv.org/abs/2406.06608