code: def partial_argnums(f, args, dyn_argnums):
      fixed_args = [
          None if i in dyn_argnums else arg
          for i, arg in enumerate(args)
      ]
  
      def f_(*dyn_args):
          args = fixed_args[:]
          for i, arg in zip(dyn_argnums, dyn_args):
              args[i] = arg
          return f(*args)
  
      dyn_args = [args[i] for i in dyn_argnums]
      return f_, dyn_args
  
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 24 for calling a global function with example usage: partial_argnums(f, args, dyn_argnums) and returns: f_,  dyn_args, f