Companion code to https://arxiv.org/abs/2409.03797v2 - View it on GitHub