This recent paper is relevant: https://arxiv.org/abs/2412.06769