Switch from post-norm layers to pre-norm layers to improve rollout skill

[alxmrs]

Idea from @adam-subel given his experiments.

[adam-subel]

In particular, this seems to reduce the immediate regression (in mean temperature) to the training data and yields modest improvements in trends. I will add some plots when they are polished