@RohanAlexander shared this paper that might be good to reference in this manuscript: https://arxiv.org/abs/2411.16795