code: def _odeint_wrapper(func, rtol, atol, mxstep, hmax, y0, ts, *args):
      y0, unravel = ravel_pytree(y0)
      func = ravel_first_arg(func, unravel)
      out = _odeint(func, rtol, atol, mxstep, hmax, y0, ts, *args)
      return jax.vmap(unravel)(out)
  
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 11 for calling a global function with example usage: _odeint_wrapper(func, rtol, atol, mxstep, hmax, y0, ts) and returns: jax