TorchX SDK and Components
Project description
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.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
torchx-0.1.0.tar.gz
(105.9 kB
view hashes)
Built Distributions
torchx-0.1.0-py3.9.egg
(393.6 kB
view hashes)
torchx-0.1.0-py3.8.egg
(315.1 kB
view hashes)
torchx-0.1.0-py3-none-any.whl
(179.2 kB
view hashes)