TorchX¶

TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and support for E2E production ML pipelines when you’re ready.

GETTING STARTED? Follow the quickstart guide.

In 1-2-3¶

Step 1. Install

pip install torchx[dev]

Step 2. Run Locally

torchx run --scheduler local_cwd utils.python --script my_app.py "Hello, localhost!"

Step 3. Run Remotely

torchx run --scheduler kubernetes utils.python --script my_app.py "Hello, Kubernetes!"

Documentation¶

Usage

Works With¶

Schedulers

Pipelines

Kubeflow Pipelines

Examples¶

Examples

Components Library¶

Components

Runtime Library¶

Application (Runtime)

Reference¶

API

Best Practices