There is a small labeling bug in Autoregressive: the parameter dimensions are using AR_PARAM_DIM (always equal to "lag_ar"), which is inconsistent for multivariate param dims. I changed it to f"lag_{self.name}" (e.g "lag_auto_regressive") and will push the PR as soon as I can