code: def module_broadcast(m, broadcast_fn, *args, **kwargs):
      
      apply_leaf(m, lambda x: module_apply_broadcast(x, broadcast_fn, args, kwargs))
docs: Call given function in all submodules with given parameters