Slow Sampling after recent updates? PYMC MMM v5

Hi @ricardoV94, thanks for the help. Tried the batched convolution but didn’t change.

Sampler: numpyro
PYMC: 5.22.0
PYMC Marketing: 0.13.1
Python: 3.13.3 (also tried with 3.11.12 & 3.11.3)