TorchX¶
TorchX is an application spec SDK for PyTorch related apps. It defines
standard structs called specs
that represent the job definition of an
application. The application spec
is the common language between
TorchX runners
and pipeline adapters
. Once an application’s spec
is created, the application can be run as a standalone job on a cluster or
as a stage in an ML pipeline/workflow. TorchX works with several mainstream
job schedulers and ML pipeline platforms so chances are you are already familiar
with one that works with TorchX.
Just getting started? First learn the basic concepts and take a look at the builtin components library.
Not finding the component or adapter you are looking for? Write a custom one that fits your needs by using our SDK.
Documentation¶
Usage
Examples
Components Library¶
Components
Works With¶
Schedulers
Pipeline Adapters
Experimental¶
Beta Features