Theano TypeError during model building with celerite2 (pymc3)

Question

I recently switched from celerite to celerite2 for modelling stellar light curves. I have followed the celerite 2 tutorial closely, adapting it to my needs. My model consist