Complex numbers are also going to be hard in base pytensor. You will have to treat them as 2d vectors, because you can’t get complex gradients by default, see here for some discussion. JAX does handle them though, so it’s surprising to me that their gamma function can’t handle them.You might be best off with option 1 as a first pass.