Fix `wrap_jax` when there is a mix of statically known and unknown shapes#1722

ricardoV94 merged 1 commit intopymc-devs:mainpymc-devs/pytensor:mainfrom

ricardoV94:wrap_jax_bugricardoV94/pytensor:wrap_jax_bugCopy head branch name to clipboard