code: def _f1(params, x):
      return x * tf.reduce_mean(params**2) + 1.0
  
imports: from absl.testing import absltest
  from absl.testing import parameterized
  import jax
  from jax import numpy
  import neural_tangents
  from neural_tangents import experimental
  import numpy
  import tensorflow
label_desc: Use image node_id 1 for calling a global function with example usage: _f1(params, x) and returns: unknown