Fix bug in `fit_MAP` when shared variables are used in graph#468

jessegrabowski merged 4 commits intopymc-devs:mainpymc-devs/pymc-extras:mainfrom

jessegrabowski:shared-vars-map-jaxjessegrabowski/pymc-extras:shared-vars-map-jaxCopy head branch name to clipboard