jax.experimental.host_callback.CallbackException

exception jax.experimental.host_callback.CallbackException

表示某些回调函数出现异常。

由 barrier_wait() 抛出。有关详细信息，请参阅 jax.experimental.host_callback 模块文档。