code: def for_body_sincostan(i, refs):
      a_ref, b_ref = refs
      a = a_ref[i]
      b_ref[i] = jnp.tan(jnp.sin(jnp.cos(a)))
  
imports: from functools import partial
  from absl.testing import absltest
  from absl.testing import parameterized
  import numpy
  import jax
  from jax import random
  from jax._src import test_util
  from jax._src.lax.control_flow import for_loop
  import jax.numpy
  from jax import config
label_desc: Use image node_id 8 for calling a global function with example usage: for_body_sincostan(i, refs) without return types