Consider initialising `UserDataInit` in two stages

[ray-chew]

The first stage initialises the variables based on the user_data dependency graph.
The second stage then initialises everything else based on the variables initialised in the first stage, e.g., the stratification function etc.