code: def defvjp_argnums(fun, vjpmaker):
      primitive_vjps[fun] = vjpmaker
  
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 3 for calling a global function with example usage: defvjp_argnums(fun, vjpmaker) without return types