After multiple attempts, calculating the variance of states.particles still scales very poorly. An example is shown below. Profiling shows the slowdown comes from MPI.Reduce!. It needs to be investigated if anything can be done about this.