jax.experimental.host_callback.CallbackException#
- exception jax.experimental.host_callback.CallbackException[source]#
Signals that some callback function had exceptions.
Raised by
barrier_wait()
. See thejax.experimental.host_callback
module documentation for details.