jax.experimental.host_callback.CallbackException

exception jax.experimental.host_callback.CallbackException

信号表示某些回调函数发生了异常。

由 barrier_wait() 引发。详情请参阅 jax.experimental.host_callback 模块文档。