code: def defvjp_argnum(fun, vjpmaker):
      def vjp_argnums(argnums, *args):
          vjps = [vjpmaker(argnum, *args) for argnum in argnums]
          return lambda g: (vjp(g) for vjp in vjps)
  
      defvjp_argnums(fun, vjp_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 4 for calling a global function with example usage: defvjp_argnum(fun, vjpmaker) and returns: function