I’m using the 4.0 beta from GitHub for this, I figured if I was going to go deep in the weeds on Theano i might as well migrate straight to Aesara.
The only downside to this method is that since I’m using normal conjugacy to compute the likelihood, you have to stick with normally distributed errors.