Try using plot_forest for vector-valued variables. plot_trace is really just for inspecting individual parameters for convergence, etc. You can restrict the variables plotted with any of the plotting functions using the var_names argument.
1 Like