jax.experimental.host_callback.CallbackException#

exception jax.experimental.host_callback.CallbackException[source]#

Signals that some callback function had exceptions.

Raised by barrier_wait(). See module documentation for details.