Improve numpyro output format
In open MRs numpyro results are just dumped to an arviz InferenceData object which is not compatible with the plotting/common format/pesummary jobs.
This might naturally interact with converting to the popsummary format.