Compatibility with `jaxtyping` about quax HOT 1 OPEN

This looks expected to me. Once a Value passes through a quaxify boundary, then it should look like an array to everything inside.

Put another way, the underlying function should be one that acts on arrays. It is by wrapping it in a quaxify that it becomes able to consume Values.

If you use the jaxtyping import hook, then the type-checking is always put on the bottom of the decorator list (this is needed for compatibility with jax.custom_{jvp,vjp}. In particular, that means it happens inside the quaxify.

from quax.