TorchX SDK and Components
Project description
to install fork
pip install torchx-applovin
to update
rm dist/* && && python3.9 setup.py sdist bdist_wheel && twine upload dist/* --verbose
TorchX
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:
- Kubernetes (EKS, GKE, AKS, etc)
- Slurm
- AWS Batch
- Docker
- Local
- Ray (prototype)
- GCP Batch (prototype)
Need a scheduler not listed? Let us know!
Quickstart
See the quickstart guide.
Documentation
Requirements
torchx:
Certain schedulers may require scheduler specific requirements. See installation for info.
Installation
Stable
# 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]"
Nightly
# install torchx sdk and CLI
pip install torchx-nightly[dev]
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[dev]
Docker
TorchX provides a docker container for using as as part of a TorchX role.
See: https://github.com/pytorch/torchx/pkgs/container/torchx
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
Built Distribution
Hashes for torchx-applovin-0.6.0.dev12.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | 016dadb01266fbdc21f7a736c9c0a0d4eea8241ff7fb5e5d077f6dc471fd196e |
|
MD5 | 7b884f0b7a490ae3637a184e328e3d5c |
|
BLAKE2b-256 | b0d53912f970b7b7990cc626f1557ca3e2b2be6e0278d3b7b2ae30750fe768b1 |
Hashes for torchx_applovin-0.6.0.dev12-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 9f64bf90fd9a3fcbdc0e76080acd1de9e415baa824a7c4e6d8d7d0c4535e537b |
|
MD5 | a2d56180a3582365f6b1735edf09fad3 |
|
BLAKE2b-256 | 48a6a2667af87795be71a0b0d2c9e74840b99922f7c9a8ffee16b8d5419a1987 |