Porting stan model to PyMC

We could also implement a DiffTransform to the same effect and hope/check if Aesara is clever enough to skip a diff of a cumsum