TODO: - [ ] Add timers for key functions of the MultiStateSampler - [ ] Run benchmarks looking at where we are spending most of the time
