jax.experimental.host_callback.CallbackException
Warning
This page was created from a pull request (#9655).
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.