TorchX
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.
Requirements
TorchX SDK (torchx):
- python3 (3.8+)
- torch
TorchX Kubeflow Pipelines Support (torchx-kfp):
- torchx
- kfp
Installation
Release
# 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]"
Nightly
# install torchx sdk and CLI
pip install torchx-nightly
Source
# 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]
Quickstart
See the quickstart guide.
Contributing
We welcome PRs! See the CONTRIBUTING file.
License
TorchX is BSD licensed, as found in the LICENSE file.