ESS in arviz and numpyro

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).