write energy as an attribute

[yunfanzhou]

Energies can be expensive to compute and should therefore sometimes be cached. There are also other reasons to look into this (such as caching for comparing old and new states, which is being looked into by #567)