def METHOD_NAME(model, args, rng_key, X, Y):
    start = time.time()
    kernel = NUTS(model)
    mcmc = MCMC(
        kernel,
        num_warmup=args.num_warmup,
        num_samples=args.num_samples,
        num_chains=args.num_chains,
        progress_bar=False if "NUMPYRO_SPHINXBUILD" in os.environ else True,
    )
    mcmc.run(rng_key, X, Y)
    mcmc.print_summary(exclude_deterministic=False)
    samples = mcmc.get_samples()
    summary_dict = summary(samples, group_by_chain=False)
    print("\nMCMC elapsed time:", time.time() - start)
    return summary_dict