Plotting a trace potentially causes my Python kernel to crash?

Can you transform your trace to InferenceData?

idata = az.from_pymc3(trace)