Fix wrap_jax when there is a mix of statically known and unknown shapes
#5561
This job was skipped
Loading
wrap_jax when there is a mix of statically known and unknown shapes
#5561