ODE_UY gradient checks now kernel unit