MLJ and DiffEqFlux

Hello everyone,

I would like to know if it is possible to use MLJFlux with DiffEqFlux, or does it exist a interface?

Do I have to add method, such as:

fit!(loss, penalty, chain::Something_related_to_DiffEqFlux, optimiser, epochs, verbosity, X, y)

#appropriate code to DiffEqFlux training neural ODE



MLJFlux does not currently interface DiffEqFlux. I’m guessing that more than your suggestion would be involved, but I’m probably not familiar enough with DiffEqFlux to be sure.