fix(gh-2036): MyPy Errors in Distributions Module#2050

Qazalbash wants to merge 3 commits intopyro-ppl:masterpyro-ppl/numpyro:masterfrom

Qazalbash:issue-2036Qazalbash/numpyro:issue-2036Copy head branch name to clipboard