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!"