AWS和Facebook加入了PyTorch的新开源项目

亚马逊网络服务公司(AmazonWe bServicesInc.)和Face book公司(Face bookInc.)今天联合宣布了几个新的开源项目,用于培训人工智能模型的Py Torch是一个流行的开源机器学习框架。

Py Torch是由Face book的AI研究小组创建的,作为编程语言Python的函数机器学习库。 它主要用于深度学习,这是机器学习的一个分支,试图模仿人脑的运作方式,并在语言翻译、图像和语音识别等领域取得了重大突破。

Py Torch应该有助于加快这些AI能力的发展,并且以前曾被用于为Face book的Oculus虚拟现实耳机构建更真实的化身。 例如,加州大学伯克利分校的研究人员也利用Py Torch来加快他们在图像到图像转换方面的工作。

今天宣布的新的Py火炬项目包括火炬服务,这是Py火炬的一个模型服务框架,使开发人员更容易将新投入生产。 第二个是TorchElastic,它是一个库开发人员可以用来在Kubernetes集群上构建容错培训作业,例如Amazon的EC2点实例或其AWS弹性Kubernetes服务。

根据亚马逊的一篇博客文章,火炬服务库支持用Python和火炬脚本编程语言编写的模型。 主要的好处是它使开发人员能够同时运行一个模型的多个版本,甚至回滚到该模型的前一个版本。

至于TorchElastic,这使用户能够根据自己的需要,扩大他们基于云的AI模型培训资源。 这些公司说,它应该用于大型分布式机器学习项目,如自然语言处理和计算机视觉。

Face book的博客文章写道:“Kubernetes和TorchElastic的集成允许Py Torch开发人员在一组计算节点上训练机器学习模型,这些节点可以在不干扰培训工作的情况下动态变化。 “火炬弹性的内置容错能力允许培训继续进行,即使节点在培训过程中下降。 这可以采取服务器维护事件、网络问题或服务器节点抢占等形式。

同时,Py Torch1.5版本包括一个稳定的C前端应用程序编程接口,该接口使框架能够将模型从PythonAPI转换为C API。

这些公司说,py火炬1.5还配备了升级的火炬视觉、火炬文本和火炬音频图书馆。 使Py Torch能够与GoogleCloud Tensor处理单元芯片一起使用的torch_xla包也已更新。 去年10月在旧金山举行的年度Py火炬开发者大会上,Face book首次为GoogleCloud TPU提供了支持。