code: def fast_color_perturb(input_image, aug_params):
      return jnp.clip(input_image + aug_params.reshape(1, 1, 3), -1, 1)
  
imports: import jax
  import jax.numpy
label_desc: Use image node_id 3 for calling a global function with example usage: fast_color_perturb(input_image, aug_params) and returns: jnp