Graph Neural Ode Python