I suspect your computational graph is too large/inefficient and compilation as well as function evaluation takes a prohibitively long time.
My first guess would be to do away with those python loops, and try to think of a way to achieve the same with numpy-like vectorization or if not possible, use a Pytensor scan, which defines symbolic loops without growing the graph in the number of iterations