I get different effective sample size between arviz.ess (default) and print_summary(). What definition are we using in numpyro?
https://arviz-devs.github.io/arviz/api/generated/arviz.ess.html
We follow Stan manual for the implementation. I guess arviz uses an updated version. Probably numpyro is equivalent to arviz with method="mean" (I’m not sure, you can try it out).