Perhaps @junpenglao knows this? I am a bit out of touch with PyMC V3, so I don’t know it from the top of my head.