JAX

• JAX 插件 API 参考
  • DALIGenericIterator
  • data_iterator()
• JAX 框架
  • JAX 和 DALI 入门
  • 使用 DALI 和 JAX 训练神经网络
  • 使用多个 GPU 进行训练
  • 使用 DALI 和 Flax 训练神经网络
  • 带有 DALI 和 FLAX 的多 GPU
  • 使用 DALI 和 Paxml 训练神经网络
  • T5X 和 DALI 入门
• 在 DALI pipeline 中运行 JAX