Call given function in all submodules with given parameters

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