Code for https://arxiv.org/pdf/2608.16157 - View it on GitHub