torchx (original) (raw)
Project description
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.
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
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file torchx-0.7.0.tar.gz.
File metadata
- Download URL: torchx-0.7.0.tar.gz
- Upload date: Jul 16, 2024
- Size: 196.1 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.1 CPython/3.8.10
File hashes
Hashes for torchx-0.7.0.tar.gz | Algorithm | Hash digest | | | ----------- | ---------------------------------------------------------------- | | | SHA256 | 933b800849c69ddff9feda931f0ae2c4083638eafc73abfdf361158667c68ad6 | | | MD5 | 8813dee85f6c44e94d3e20310ba7ccb4 | | | BLAKE2b-256 | eda9e07e64222e59f44eb18b333bc8db942a295c0d9c16f04c8703fe106a8fd5 | |
See more details on using hashes here.
File details
Details for the file torchx-0.7.0-py3-none-any.whl.
File metadata
- Download URL: torchx-0.7.0-py3-none-any.whl
- Upload date: Jul 16, 2024
- Size: 256.1 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.1 CPython/3.8.10
File hashes
Hashes for torchx-0.7.0-py3-none-any.whl | Algorithm | Hash digest | | | ----------- | ---------------------------------------------------------------- | | | SHA256 | 815c2628c30de1f5938c14c4427cfe280fdf93473b171be906ab7c1ea5971824 | | | MD5 | ecafa8db39a531fa21483e53ed62f521 | | | BLAKE2b-256 | a08d282b884cef8f26b240e1ac8e60513a25a68e3463dc1c68771191a2c8a209 | |