Fix bug in storage_input alignment of the JAX backend#587

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

ricardoV94:fix_shared_rng_replacement_bugricardoV94/pytensor:fix_shared_rng_replacement_bugCopy head branch name to clipboard