Hi!
Did you try plotting the trace directly with ArviZ instead of PyMC: az.plot_trace(my_difficult_trace?
Also, from what I understand, you’re just sampling one parameter right? Is it multi-dimensional? If yes, this could cause the RAM saturation – then I’d try az.plot_trace(compact=True). This will plot multidimensional variables in a single plot.