Model checkpoints for our paper: https://arxiv.org/pdf/2509.22746