Robuta

https://www.tensorflow.org/probability/api_docs/python/tfp/substrates/jax/mcmc/effective_sample_size
Estimate a lower bound on effective sample size for each independent chain.
effective sample sizetfpsubstratesjaxmcmc