Heckman Selection Model

Hi Forum,

I’m trying to put together a little Heckman Sample Selection model in pyro. I’ve done this in Stan but was hoping to see if anyone has done this already and is willing to share their code.

Z