https://arxiv.org/abs/2306.11644 is along these lines.