How to choose a distribution for the likelihood step?

You can try az.plot_pair().