Draft implementation of gradient flow with Runge-Kutta.