Workarounds for `arviz` no longer supporting `MultiTrace`?

Are you on the latest version of pymc? Can you share a reproducible snippet that shows the memory leakage?

It would be surprising that switching the backend would lead to leakage