nvidia.dali.plugin.jax.fn.jax_function
--------------------------------------

.. autofunction:: nvidia.dali.plugin.jax.fn.jax_function

.. seealso::
   * `Running custom JAX augmentations in DALI <../examples/custom_operations/jax_operator_basic.html>`_
   * `Running JAX augmentations on multiple GPUs <../examples/custom_operations/jax_operator_multi_gpu.html>`_