Save the initialization state in the CSV file when doing optimization and saving warmup

This is useful for debugging io.  It also allows the user to get the log prob value for the initial state.