nvidia.dali.plugin.jax.fn

The following table lists all operations available in nvidia.dali.plugin.jax.fn module:

Function	Device support	Short description
jax_function	CPU, GPU	Transforms jax.Array processing function into DALI operator