You can try to fix your likelihood mu or sigma to a constant to see which one is causing nan.
Also you can disable jittering in the initialization of NUTS (in pm.sample) to see if that is what’s causing problems.
1 Like