jax.experimental.host_callback.rst
``jax.experimental.host_callback`` module
=========================================
.. automodule:: jax.experimental.host_callback
API
---
.. autosummary::
:toctree: _autosummary
id_tap
id_print
call
barrier_wait
CallbackException