在 DALI 管道中运行 JAX

有关作为 DALI 迭代器或管道一部分运行自定义 JAX 增强的文档和示例，请参阅 plugin.jax.fn.jax_function。