PyTorch 非常易于使用，并能够构建复杂的 AI 模型。但一旦研究变得复杂，并且将诸如多 GPU 训练，16 位精度和 TPU 训练之类的东西混在一起，用户很可能会引入错误。PyTorch Lightning 完全解决了这个问题。Lightning 会构建用户的 PyTorch 代码，以便可以抽象出培训的细节。这使得人工智能研究可扩展且可快速迭代。点击下方链接查看使用教程。