JAX 框架

• JAX 和 DALI 入门
  • 前提条件
  • 快速开始
  • 使用 JAX 运行 DALI
  • 批处理
  • GPU 加速
  • 硬件加速解码
  • 随机洗牌
  • 增强
  • 多个 GPU
  • 技术细节
  • 下一步去哪里
• 使用 DALI 和 JAX 训练神经网络
• 使用多个 GPU 进行训练
  • 使用自动并行化进行训练
  • 使用 pmapped 迭代器进行训练
• 使用 DALI 和 Flax 训练神经网络
• 具有 DALI 和 FLAX 的多个 GPU
• 使用 DALI 和 Paxml 训练神经网络
• T5X 和 DALI 入门
  • 前提条件
  • 用于 T5X 的可预览迭代器
  • 下一步去哪里