Running JAX in DALI pipeline
============================

See the :meth:`plugin.jax.fn.jax_function <nvidia.dali.plugin.jax.fn.jax_function>` for
documentation and examples on running custom JAX augmentations as a part of DALI
iterator or pipeline.