Thanks for the nice package.
Is there anyway to convert
print_summary() to pandas data frame.
I can make an arviz object but for large problems it takes actually longer than my sampling .
numpyro.diagnostics.summary(mcmc.get_samples(group_by_chain=True)) took as long as using
mcmc.print_summary() is printing almost instantly.
summary under the hood.