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
Metadata
3 maintainers
Release files for torchx-nightly 2026.10.10
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.10.10-py3-none-any.whl | Python 3 | none | any | Details |
Release files / torchx_nightly-2026.10.10-py3-none-any.whl
| Download URL | torchx_nightly-2026.10.10-py3-none-any.whl |
|---|---|
| Size | 239.1 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
d76109d44d185c1189d1fe0712c6d6a233b0aeb9d38c00c20c82765f85d25738
|
|
BLAKE2b-256 checksum How to use checksums |
16e10ff6c5c8bee0c48e0a2f720f6714ec99d44bf4d7954f6f96565194d661c1
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/7.0.0 CPython/3.12.3
|
