backward                Backward pass of the Neural ODE
forward                 Forward pass of the Neural ODE network
rk4_step                Runge Kutta solver for ordinary differential
                        equations
