I get different effective sample size between arviz.ess (default) and print_summary(). What definition are we using in numpyro?
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).