Looks good! Thank you! Should we also add instrumentation to method like evaluate_main_transition(), evaluate_aux_transistion(), evaluate_lagranage_transition() (e.g., here).
That would log the function each time it is called, which is not what we want. Probably there is a better way to get the overall time instead.
That would log the function each time it is called, which is not what we want. Probably there is a better way to get the overall time instead.