jax.experimental.checkify.JaxRuntimeError

exception jax.experimental.checkify.JaxRuntimeError