improve docs and error message for odeint *args (#2931)
improve docs and error message for odeint *args (#2931)

cf. #2920
google/jax,tensorflow/probability,google/jax,tensorflow/probability,google/jax,google/jax