Fix wrap_jax when there is a mix of statically known and unknown shapes#1722
Open
ricardoV94 wants to merge 1 commit intopymc-devs:mainfrom
Open
Fix `wrap_jax` when there is a mix of statically known and unknown shapes#1722ricardoV94 wants to merge 1 commit intopymc-devs:mainfrom
ricardoV94 wants to merge 1 commit intopymc-devs:mainfrom