code: def inner_prod(xs, ys):
      xys = zip(xs, ys)
      assert all(x.shape == y.shape for x, y in xys)
      return sum(jnp.sum(x * y) for x, y in xys)
  
imports: from collections import namedtuple
  from functools import partial
  import numpy.random
  from absl.testing import absltest
  from absl.testing import parameterized
  import itertools
  import jax.numpy
  from jax import jit, jvp, vjp
  import jax._src.test_util
  from jax import config
  from jax._src.util import unzip2, safe_zip, safe_map
label_desc: Use image node_id 20 for calling a global function with example usage: inner_prod(xs, ys) and returns: sum