`jax.experimental.host_callback` module

.. automodule:: jax.experimental.host_callback

API

.. autosummary::
   :toctree: _autosummary

   id_tap
   id_print
   call
   barrier_wait
   CallbackException