"PyTorch Lightning 每月下載量達(dá)400萬次。"

作者：Darin

編輯：tuya

近日，開源軟件PyTorch Lightning framework的公司Lightning AI宣布，它在Coatue領(lǐng)投的B輪融資中募集了4000萬美元，參投方還包括Index Ventures、貝恩資本、Chainsmokers的 Mantis VC和First Minute Capital。

本輪資金將用于擴(kuò)大團(tuán)隊規(guī)模，以及支持PyTorch Lightning主題社區(qū)。

Lightning AI由公司首席執(zhí)行官William Falcon和福布斯前數(shù)據(jù)產(chǎn)品負(fù)責(zé)人Luis Capelo共同于2019年創(chuàng)立。William Falcon在紐約大學(xué)和Facebook AI Research攻讀博士期間，創(chuàng)建了開源神經(jīng)網(wǎng)絡(luò)開發(fā)工具PyTorch Lightning.

Lightning AI銷售的PyTorch Lightning商業(yè)版是同類產(chǎn)品中最受歡迎的技術(shù)之一，該產(chǎn)品每月下載量達(dá)400萬次。

PyTorch Lightning之所以在開發(fā)人員中受歡迎，是因為它簡化了開發(fā)AI軟件所涉及的許多手動任務(wù)。開發(fā)一個AI應(yīng)用程序，不僅需要構(gòu)建神經(jīng)網(wǎng)絡(luò)，還需要設(shè)置和管理運(yùn)行神經(jīng)網(wǎng)絡(luò)的基礎(chǔ)設(shè)施，維護(hù)基礎(chǔ)設(shè)施需要大量的手動工作，這會減慢機(jī)器學(xué)習(xí)項目的速度。

PyTorch Lightning可簡化的任務(wù)之一，是AI訓(xùn)練。大規(guī)模的神經(jīng)網(wǎng)絡(luò)通常過于復(fù)雜，無法使用單個處理器進(jìn)行訓(xùn)練，因此使用多個芯片，這在技術(shù)上可能具有挑戰(zhàn)性。

PyTorch Lightning還可使開發(fā)人員更容易評估神經(jīng)網(wǎng)絡(luò)速度。此外，如果在訓(xùn)練神經(jīng)網(wǎng)絡(luò)時出現(xiàn)技術(shù)問題，PyTorch Lightning會暫停并執(zhí)行故障排除流程。

軟件團(tuán)隊通常使用圖形處理單元構(gòu)建AI軟件。但是，在某些情況下，機(jī)器學(xué)習(xí)項目可能還需要使用其他類型的芯片，例如Google LLC使用了的是公共云提供的AI優(yōu)化張量處理單元（TPU）。PyTorch Lightning減少了開發(fā)人員因此必須編寫的自定義代碼量。

Lightning AI通過PyTorch Lightning的商業(yè)托管版本產(chǎn)生收入，使企業(yè)更容易使用該工具。此外，常用的應(yīng)用場景中，預(yù)打包的AI應(yīng)用程序也在PyTorch Lightning功能中?？蛻羰褂肔ightning的應(yīng)用程序，就能避免必須從頭部署所有內(nèi)容。

Falcon表示，“到目前為止，還沒有一種生產(chǎn)級AI應(yīng)用程序，能將開發(fā)到生產(chǎn)的整個過程考慮在內(nèi)。Lightning AI奉行簡單、模塊化和可持續(xù)性的開發(fā)原則，并應(yīng)用于整個AI開發(fā)和基礎(chǔ)設(shè)施生命周期?！?/p>