mirror of
https://github.com/ROCm/jax.git
synced 2025-04-17 12:26:07 +00:00
21 lines
257 B
ReStructuredText
21 lines
257 B
ReStructuredText
``jax.experimental.host_callback`` module
|
|
=========================================
|
|
|
|
|
|
.. automodule:: jax.experimental.host_callback
|
|
|
|
API
|
|
---
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
id_tap
|
|
id_print
|
|
call
|
|
barrier_wait
|
|
CallbackException
|
|
|
|
|
|
|