Energy calculation uses too much GPU memory

[leios]

In the process of re-writing the energy calculation, we needed a bit more memory than previously thought. As such, if we are running a memory-intensive task, the energy calculation will not work correctly. We need to find a way to rewrite it to minimize the memory footprint.