code: def defjvp_argnum(fun, jvpmaker):
      def jvp_argnums(argnums, gs, ans, args, kwargs):
          return sum_outgrads(
              jvpmaker(argnum, g, ans, args, kwargs)
              for argnum, g in zip(argnums, gs)
          )
  
      defjvp_argnums(fun, jvp_argnums)
  
imports: from itertools import count
  from functools import reduce
  from .tracer import trace, primitive, toposort, Node, Box, isbox, getval
  from .util import func, subval
  import warnings
label_desc: Use image node_id 9 for calling a global function with example usage: defjvp_argnum(fun, jvpmaker) and returns: sum_outgrads