Implement a neural ode
Implement a neural ode