Numpyro sampler but do not return InferenceObject

The way forward is InferenceData so we didn’t offer the old MultiTrace for new samplers, to also reduce code complexity. Is it too much work to update the plotting functionality?