Amazon Web Services Inc.和Facebook Inc.今天共同宣布了PyTorch的幾個新的開源項(xiàng)目，PyTorch是一種流行的開源機(jī)器學(xué)習(xí)框架，用于訓(xùn)練人工智能模型。

PyTorch是由Facebook的AI研究小組創(chuàng)建的，它是針對編程語言Python的機(jī)器學(xué)習(xí)功能庫。它主要設(shè)計(jì)用于深度學(xué)習(xí)，深度學(xué)習(xí)是機(jī)器學(xué)習(xí)的一個分支，旨在模仿人腦的運(yùn)作方式，并在語言翻譯，圖像和語音識別等領(lǐng)域取得了重大突破。

PyTorch應(yīng)該有助于加速這類AI功能的開發(fā)，并且以前已被用來為Facebook的Oculus虛擬現(xiàn)實(shí)耳機(jī)構(gòu)建更逼真的頭像。例如，加州大學(xué)伯克利分校的研究人員還使用PyTorch加快了圖像到圖像轉(zhuǎn)換的工作。

今天宣布的新PyTorch項(xiàng)目包括TorchServe，這是PyTorch的模型服務(wù)框架，使開發(fā)人員更容易將新模型投入生產(chǎn)。第二個是TorchElastic，該庫是開發(fā)人員可以用來在Kubernetes集群(例如Amazon的EC2競價(jià)型實(shí)例或其AWS Elastic Kubernetes服務(wù))上構(gòu)建容錯培訓(xùn)工作的庫。

根據(jù)亞馬遜的博客文章，TorchServe庫支持以Python和TorchScript編程語言編寫的模型。它的主要好處是使開發(fā)人員可以同時運(yùn)行模型的多個版本，甚至可以回滾到該模型的先前版本。

對于TorchElastic，這使用戶可以根據(jù)自己的需求擴(kuò)展其基于云的AI模型培訓(xùn)資源。這些公司表示，它打算用于大型分布式機(jī)器學(xué)習(xí)項(xiàng)目，例如自然語言處理和計(jì)算機(jī)視覺。

Facebook的博客文章說：“ Kubernetes和TorchElastic的集成使PyTorch開發(fā)人員可以在可以動態(tài)更改而不會中斷訓(xùn)練工作的計(jì)算節(jié)點(diǎn)集群上訓(xùn)練機(jī)器學(xué)習(xí)模型。” “ TorchElastic內(nèi)置的容錯功能使訓(xùn)練可以繼續(xù)進(jìn)行，即使在訓(xùn)練過程中節(jié)點(diǎn)出現(xiàn)故障也是如此。這可以采取諸如服務(wù)器維護(hù)事件，網(wǎng)絡(luò)問題或搶占服務(wù)器節(jié)點(diǎn)之類的形式。”

同時，PyTorch 1.5發(fā)行版包含一個穩(wěn)定的C ++前端應(yīng)用程序編程接口，該接口使該框架能夠?qū)⒛Ｐ蛷腜ython API轉(zhuǎn)換為C ++ API。

兩家公司表示，PyTorch 1.5還帶有升級的torchvision，torchtext和torchaudio庫。使PyTorch可以與Google Cloud Tensor Processing Unit芯片一起使用的torch_xla軟件包也已更新。在去年10月于舊金山舉行的年度PyTorch開發(fā)者大會上，F(xiàn)acebook首先添加了對Google Cloud TPU的支持。