One general comment about sca0, d0, feq0, etc -- I'd prefer we just drop the index instead of adding grid_idx everywhere. I'm 95% sure this is now possible, but please correct me if I'm wrong and missed something. The variables are named this way, because at some point in the past we used to load variables from multiple lattices into memory at the same time.