Transitioning from pm.sample() to pm.fit()

The nan happens in the optimization, because it gets too extreme gradients probably