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
pip install torchx
# install torchx kubeflow pipelines (kfp) support
pip intall torchx[kfp]See the quickstart guide.
We welcome PRs! See the CONTRIBUTING file.
TorchX is BSD licensed, as found in the LICENSE file.