def METHOD_NAME(expr, rng=None):
    """
    Make all of the stochastic nodes in expr use the rng
    uniform(0, 1) -> uniform(0, 1, rng=rng)
    """
    if rng is None:
        rng = np.random.default_rng()
    lrng = as_apply(rng)
    for node in dfs(expr):
        if node.name in implicit_stochastic_symbols:
            for ii, (name, arg) in enumerate(list(node.named_args)):
                if name == "rng":
                    node.named_args[ii] = ("rng", lrng)
                    break
            else:
                node.named_args.append(("rng", lrng))
    return expr