Trace plots Hierarchical Models

You can also just update PyMC to 3.9.3 from pip or conda-forge