code: def for_body_sin_sq(i, refs):
      x_ref, y_ref = refs
      x = x_ref[i]
      y = x
      y_ref[i] = y
      y = y_ref[i]
      y_ref[i] = jnp.sin(y * y)
  
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 11 for calling a global function with example usage: for_body_sin_sq(i, refs) without return types