def single_sample_fn(rng):
    return sample_posterior_predictive(rng, model, model_args, guide, guide_args, params, with_intermediates=with_intermediates, **kwargs)
def single_sample_fn(rng):
    return sample_posterior_predictive(rng, model, model_args, guide, guide_args, params, with_intermediates=with_intermediates, **kwargs)