code: def conv_args(k, f):
      return dict(
          w_init=functools.partial(
              objax.random.normal,
              stddev=objax.functional.rsqrt(0.5 * k * k * f),
          )
      )
  
file_path: objax/examples/fixmatch/libml/zoo/resnet.py
imports: import functools
  from typing import Callable
  import jax
  import objax
  from objax.typing import JaxArray
label_desc: Use image node_id 2 for calling a global function with example usage: conv_args(k, f) and returns: dict