As a hack you could generate the chains independently
traces = [pm.sample(chains=1) for _ in xrange(n_chains)]