This is the state of the code at the time of writing the paper: https://arxiv.org/abs/2402.14846
It enables to recreate these experiments.