[staging] ✨ Store bookkeeping information in result or dataset

[jsnel]

The bookkeeping information (such the grouping of the global axis) which is typically only avaialble during the optimization process, but not typically at the dataset level, should also be stored somewhere.

Also look at other information which needs to be retained to fully reconstruct the optimization from the information in the datasets.