TorchX /AMP#

参考:TorchX

TorchX 用于从研发到生产快速构建和部署 ML 应用程序的 SDK。它提供了各种编码 MLOps 最佳实践的内置组件,并使所有人都可以访问诸如分布式训练和超参数优化等高级特性。

用户可以在没有额外设置成本的情况下开始使用 TorchX,因为它支持已经在生产中广泛采用和部署的流行 ML 调度器和管道编排器。没有两个生产环境是相同的。为了满足各种用例,TorchX 的核心 API 允许在定义良好的扩展点上进行大量的定制,因此即使是最独特的应用程序也可以在不定制整个垂直堆栈的情况下得到服务。

混合精度