Shortcuts

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

Works With

Examples

Components Library

Runtime Library

Application (Runtime)

Reference

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources