the pm.math library just aliases pytensor.tensor, so this will not be the source of performance issues.
If colab is running significantly faster than local there’s likely an installation issue. Did you follow the official instructions when you installed?