TorchX¶

TorchX 是一个针对 PyTorch 相关应用程序的应用程序特定 SDK。它定义了称为 specs 的标准结构体，用于表示应用程序的作业定义。应用程序 spec 是 TorchX runners 和管道 adapters 之间的通用语言。一旦创建了应用程序的 spec，应用程序可以作为集群上的独立作业运行，也可以作为机器学习管道/工作流程中的阶段。TorchX 可以与多个主流作业调度器和机器学习管道平台一起工作，因此你很可能已经熟悉其中一个与 TorchX 兼容的平台。