code: def _rotate_vectors(H, i, cs, sn):
      x1 = H[i]
      y1 = H[i + 1]
      x2 = cs.conj() * x1 - sn.conj() * y1
      y2 = sn * x1 + cs * y1
      H = H.at[i].set(x2)
      H = H.at[i + 1].set(y2)
      return H
  
imports: from functools import partial
  import operator
  import numpy
  import jax
  import jax.numpy
  from jax import device_put
  from jax import lax
  from jax import scipy
  from jax.tree_util import tree_leaves, tree_map, tree_structure, tree_reduce, Partial
  from jax._src import dtypes
  from jax._src.lax import lax
  from jax._src.util import safe_map
label_desc: Use image node_id 18 for calling a global function with example usage: _rotate_vectors(H, i, cs, sn) and returns: H