jax.experimental.host\_callback.barrier\_wait
=============================================

.. currentmodule:: jax.experimental.host_callback

.. autofunction:: barrier_wait