Runge Kutta Integration

Fourth order Runge-Kutta method for time-invariant ODEs.

jax_agents.common.runge_kutta.runge_kutta(dynamics_f, state, action, dt)

dynamics_f(state, action) -> state_dot.