code: def _odeint_fwd(func, rtol, atol, mxstep, hmax, y0, ts, *args):
      ys = _odeint(func, rtol, atol, mxstep, hmax, y0, ts, *args)
      return ys, (ys, ts, args)
  
imports: from functools import partial
  import operator
  import jax
  import jax.numpy
  from jax._src import core
  from jax import custom_derivatives
  from jax import lax
  from jax._src.numpy.util import promote_dtypes_inexact
  from jax._src.util import safe_map, safe_zip
  from jax.flatten_util import ravel_pytree
  from jax.tree_util import tree_leaves, tree_map
  from jax._src import linear_util
label_desc: Use image node_id 13 for calling a global function with example usage: _odeint_fwd(func, rtol, atol, mxstep, hmax, y0, ts) and returns: ys,  ys,  ts,  args