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.
TorchX currently supports:
- Kubernetes (EKS, GKE, AKS, etc)
- Slurm
- Docker
- Local
Need a scheduler not listed? Let us know!
Quickstart
See the quickstart guide.
Documentation
Requirements
torchx:
Certain schedulers may require scheduler specific requirements. See installation for info.
Installation
Stable
# install torchx sdk and CLI -- minimum dependencies
pip install torchx
# install with all dependencies
pip install "torchx[dev]"
# install torchx Kubernetes / Volcano support
pip install "torchx[kubernetes]"
# install torchx GCP Batch support
pip install "torchx[gcp_batch]"
Nightly
# install torchx sdk and CLI
pip install torchx-nightly[dev]
Source (Development)
# clone the repo
git clone https://github.com/meta-pytorch/torchx.git
cd torchx
# install with uv (recommended for development)
uv sync --extra dev
# or with pip
pip install -e ".[dev]"
Docker
TorchX provides a docker container image at ghcr.io/pytorch/torchx for use
as part of a TorchX role.
Contributing
We welcome PRs! See the CONTRIBUTING file.
License
TorchX is BSD licensed, as found in the LICENSE file.
3 maintainers
Release files for torchx-nightly 2026.9.20
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| torchx_nightly-2026.9.20-py3-none-any.whl | Python 3 | none | any | Details |
Release files / torchx_nightly-2026.9.20-py3-none-any.whl
| Download URL | torchx_nightly-2026.9.20-py3-none-any.whl |
|---|---|
| Size | 238.2 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
f502eecdcce16d8b7704596a9dc8ec588a37bdbc77230a4a037db2b6e7cd85c4
|
|
BLAKE2b-256 checksum How to use checksums |
d303facfea2675af7e736ee933f2e3f5c77a8d6f82df37a672b4ae14444e1bbe
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/7.0.0 CPython/3.12.3
|
