Fix error when passing `coords` and `dims` in `sampling_jax`#5983

twiecki merged 4 commits intopymc-devs:mainpymc-devs/pymc:mainfrom

bherwerth:sampling-jax-fix-idata-kwargsbherwerth/pymc:sampling-jax-fix-idata-kwargsCopy head branch name to clipboard