jax.experimental.host_callback.CallbackException

exception jax.experimental.host_callback.CallbackException

Signals that some callback function had exceptions.

Raised by barrier_wait(). See the jax.experimental.host_callback module documentation for details.