A simple solution would be to store the axis type in the parameter object and convert the state vector to a ComponentVector inside the dynamics function.
A simple solution would be to store the axis type in the parameter object and convert the state vector to a ComponentVector inside the dynamics function.