Kernel Error while running code on GPU

Specific Domains GPU
10

Did you also convert model_params, the other CPU vector?
In any case, this looks more like an issue with the use of DiffEq or Flux, so you may be better off asking people who are familiar with those packages.