NUTS sampler gets "bad initial energy" only with parallel sampling

It may be because the parallel sampling has multiple different starting points.

Try adding init='adapt_diag' when you call pm.sample . See here for more.