Seems like it is a multi-chain sampling problem - try doing trace = pm.sample(..., cores=1)
trace = pm.sample(..., cores=1)