tchatow

Here's a simple example ``` N, = jax.export.symbolic_shape("N") f = jax.export.export(jax.jit(jnp.digitize)) shape0 = jax.ShapeDtypeStruct((10,), jnp.int32) shape1 = jax.ShapeDtypeStruct((N,), jnp.int32) f(shape0, shape1) ```

@jakevdp I also observe `np.linalg.inv` polymorphism works in 0.4.31 but regressed in main. Would this be a bug or feature request? (Regression in 7b415834145df96f0c80e5cf3cb34cfe796f85b6)