Fix JAX dispatch for multi-output Composite#123

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

ricardoV94:fix_jax_composite_multi_outricardoV94/pytensor:fix_jax_composite_multi_outCopy head branch name to clipboard