Quick update. I managed to get “better” results with pm.Simulator. By better I mean that I got rhat values that are either 1.0 or very close to it without exaggerating the number of samples. Also thanks to Aesara tensor’s switch and and_ functions that allowed me to model some conditions efficiently.
1 Like