TorchX is a library containing standard DSLs for authoring and running PyTorch related components for an E2E production ML pipeline.
For the latest documentation, please refer to our website.
TorchX SDK (torchx):
- python3 (3.8+)
- torch
TorchX Kubeflow Pipelines Support (torchx-kfp):
- torchx
- kfp
# install torchx sdk and CLI -- minimum dependencies
pip install torchx
# install torchx sdk and CLI -- all dependencies
pip install "torchx[dev]"
# install torchx kubeflow pipelines (kfp) support
pip install "torchx[kfp]"
# install torchx Kubernetes / Volcano support
pip install "torchx[kubernetes]"# install torchx sdk and CLI
pip install torchx-nightly# install torchx sdk and CLI from source
$ pip install -e git+https://github.com/pytorch/torchx.git#egg=torchx
# install extra dependencies
$ pip install -e git+https://github.com/pytorch/torchx.git#egg=torchx[kubernetes]See the quickstart guide.
We welcome PRs! See the CONTRIBUTING file.
TorchX is BSD licensed, as found in the LICENSE file.