Flux, higher order derivatives and forward mode

[jw3126]

Thanks! I will see if I can make it run using sciml_train and probably continue asking questions on slack. These ZygoteRules for Dual make my head hurt