Creating a loss function for sciml_train with unequal time point distribution of the ODE components

sol[:,[1,4,5]] etc.