PyMC v5.10.3 prediction stuff

PyMC dimensions are just labels for the output of sampling. All broadcasting logic is positionally-like numpy so you need to make sure the variables are aligned correctly like you would do in numpy