Unable to load trace after VI

What error are you seeing?

As for slowness, likely because of an older version of Arviz - did you also upgraded that?