Trying to reproduce the results from https://arxiv.org/abs/2509.23886 - View it on GitHub