Function reference

All functions

backward()

Backward pass of the Neural ODE

forward()

Forward pass of the Neural ODE network

rk4_step()

Runge Kutta solver for ordinary differential equations