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