rocm_jax/docs/jax.experimental.jet.rst

mirror of https://github.com/ROCm/jax.git synced 2025-04-16 03:46:06 +00:00
``jax.experimental.jet`` module
===============================

.. automodule:: jax.experimental.jet

API
---

.. autofunction:: jet
DOC: make API doc titles more uniform 2023-01-18 10:59:42 -08:00			``jax.experimental.jet`` module
			`===============================`
Wrote docstring for jet 2022-02-12 08:14:57 +01:00
			`.. automodule:: jax.experimental.jet`

			`API`
			`---`

			`.. autofunction:: jet`