DL 框架插件

• PyTorch
  • PyTorch 插件 API 参考
  • Pytorch 框架
• TensorFlow
  • TensorFlow 插件 API 参考
  • Tensorflow 框架
• PaddlePaddle
  • PaddlePaddle 插件 API 参考
  • PaddlePaddle 框架
• JAX
  • JAX 插件 API 参考
  • JAX 框架
  • 在 DALI pipeline 中运行 JAX