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:
Need a scheduler not listed? Let us know!
See the quickstart guide.
torchx:
Certain schedulers may require scheduler specific requirements. See installation for info.
# 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 Ray support pip install "torchx[ray]" # install torchx GCP Batch support pip install "torchx[gcp_batch]"
# install torchx sdk and CLI pip install torchx-nightly[dev]
# 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[dev]
TorchX provides a docker container for using as as part of a TorchX role.
See: https://github.com/pytorch/torchx/pkgs/container/torchx
We welcome PRs! See the CONTRIBUTING file.
TorchX is BSD licensed, as found in the LICENSE file.
RetroSearch is an open source project built by @garambo | Open a GitHub Issue
Search and Browse the WWW like it's 1997 | Search results from DuckDuckGo
HTML:
3.2
| Encoding:
UTF-8
| Version:
0.7.4
