Support RandomVariable graphs with scalar shape parameters in JAX backend#1029

ricardoV94 merged 3 commits intopymc-devs:mainpymc-devs/pytensor:mainfrom

ricardoV94:jax_scalar_rv_shapesricardoV94/pytensor:jax_scalar_rv_shapesCopy head branch name to clipboard