If you die a few times over a short period of time, the chore of setting the windows back to the same spots can be a deterrent from from going for that 'one more try'.

I'd like the game to memorize the UI from the previous run.

You mean where you set lots, character stats, bags, browsers, etc?
That's not really possible.