tfp.experimental.mcmc.log_ess_from_log_weights

Computes log-ESS estimate from log-weights along axis=0.