在 DALI pipeline 中运行 JAX

请参阅 plugin.jax.fn.jax_function 文档和示例，了解如何在 DALI 迭代器或 pipeline 中运行自定义 JAX 增强。