“PyTorch Lightning 每月下载量达400万次。”

作者：Darin编辑：tuya出品：财经涂鸦（ID：caijingtuya）

近日，开源软件PyTorch Lightning framework的公司Lightning AI宣布，它在Coatue领投的B轮融资中募集了4000万美元，参投方还包括Index Ventures、贝恩资本、Chainsmokers的 Mantis VC和First Minute Capital。

本轮资金将用于扩大团队规模，以及支持PyTorch Lightning主题社区。

Lightning AI由公司首席执行官William Falcon和福布斯前数据产品负责人Luis Capelo共同于2019年创立。William Falcon在纽约大学和Facebook AI Research攻读博士期间，创建了开源神经网络开发工具PyTorch Lightning．

Lightning AI销售的PyTorch Lightning商业版是同类产品中最受欢迎的技术之一，该产品每月下载量达400万次。

PyTorch Lightning之所以在开发人员中受欢迎，是因为它简化了开发AI软件所涉及的许多手动任务。开发一个AI应用程序，不仅需要构建神经网络，还需要设置和管理运行神经网络的基础设施，维护基础设施需要大量的手动工作，这会减慢机器学习项目的速度。

PyTorch Lightning可简化的任务之一，是AI训练。大规模的神经网络通常过于复杂，无法使用单个处理器进行训练，因此使用多个芯片，这在技术上可能具有挑战性。

PyTorch Lightning还可使开发人员更容易评估神经网络速度。此外，如果在训练神经网络时出现技术问题，PyTorch Lightning会暂停并执行故障排除流程。

软件团队通常使用图形处理单元构建AI软件。但是，在某些情况下，机器学习项目可能还需要使用其他类型的芯片，例如Google LLC使用了的是公共云提供的AI优化张量处理单元（TPU）。PyTorch Lightning减少了开发人员因此必须编写的自定义代码量。

Lightning AI通过PyTorch Lightning的商业托管版本产生收入，使企业更容易使用该工具。此外，常用的应用场景中，预打包的AI应用程序也在PyTorch Lightning功能中。客户使用Lightning的应用程序，就能避免必须从头部署所有内容。

Falcon表示，“到目前为止，还没有一种生产级AI应用程序，能将开发到生产的整个过程考虑在内。Lightning AI奉行简单、模块化和可持续性的开发原则，并应用于整个AI开发和基础设施生命周期。”