Metadata-Version: 2.1
Name: torchx-nightly
Version: 2024.8.26
Summary: TorchX SDK and Components
Home-page: https://github.com/pytorch/torchx
Author: TorchX Devs
Author-email: torchx@fb.com
License: BSD-3
Keywords: pytorch,machine learning
Classifier: Development Status :: 4 - Beta
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Science/Research
Classifier: License :: OSI Approved :: BSD License
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.8
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Requires-Python: >=3.7
Description-Content-Type: text/markdown
License-File: LICENSE
Requires-Dist: pyre-extensions
Requires-Dist: docstring-parser>=0.8.1
Requires-Dist: importlib-metadata
Requires-Dist: pyyaml
Requires-Dist: docker
Requires-Dist: filelock
Requires-Dist: fsspec>=2023.10.0
Requires-Dist: urllib3<1.27,>=1.21.1
Requires-Dist: tabulate
Provides-Extra: aws_batch
Requires-Dist: boto3; extra == "aws-batch"
Provides-Extra: dev
Requires-Dist: aiobotocore==2.12.1; extra == "dev"
Requires-Dist: ax-platform[mysql]==0.2.3; extra == "dev"
Requires-Dist: boto3==1.34.51; extra == "dev"
Requires-Dist: captum>=0.4.0; extra == "dev"
Requires-Dist: docker; extra == "dev"
Requires-Dist: kubernetes==25.3.0; extra == "dev"
Requires-Dist: flake8==3.9.0; extra == "dev"
Requires-Dist: fsspec==2024.3.1; extra == "dev"
Requires-Dist: s3fs==2024.3.1; extra == "dev"
Requires-Dist: google-api-core; extra == "dev"
Requires-Dist: google-cloud-batch==0.17.14; extra == "dev"
Requires-Dist: google-cloud-logging==3.10.0; extra == "dev"
Requires-Dist: google-cloud-runtimeconfig==0.34.0; extra == "dev"
Requires-Dist: hydra-core; extra == "dev"
Requires-Dist: ipython; extra == "dev"
Requires-Dist: kfp==1.8.22; extra == "dev"
Requires-Dist: mlflow-skinny; extra == "dev"
Requires-Dist: moto~=5.0.8; extra == "dev"
Requires-Dist: pyre-extensions; extra == "dev"
Requires-Dist: pyre-check; extra == "dev"
Requires-Dist: pytest; extra == "dev"
Requires-Dist: pytest-cov; extra == "dev"
Requires-Dist: pytorch-lightning==2.3.1; extra == "dev"
Requires-Dist: tensorboard==2.14.0; extra == "dev"
Requires-Dist: sagemaker>=2.149.0; extra == "dev"
Requires-Dist: torch-model-archiver>=0.4.2; extra == "dev"
Requires-Dist: torch==2.2.1; extra == "dev"
Requires-Dist: torchmetrics==0.10.3; extra == "dev"
Requires-Dist: torchserve>=0.10.0; extra == "dev"
Requires-Dist: torchtext==0.17.1; extra == "dev"
Requires-Dist: torchvision==0.17.1; extra == "dev"
Requires-Dist: ts==0.5.1; extra == "dev"
Requires-Dist: ray[default]; extra == "dev"
Requires-Dist: lintrunner; extra == "dev"
Requires-Dist: lintrunner-adapters; extra == "dev"
Requires-Dist: grpcio==1.62.1; extra == "dev"
Requires-Dist: grpcio-status==1.48.1; extra == "dev"
Requires-Dist: googleapis-common-protos==1.63.0; extra == "dev"
Requires-Dist: google-api-core==2.18.0; extra == "dev"
Requires-Dist: protobuf==3.20.3; extra == "dev"
Provides-Extra: gcp_batch
Requires-Dist: google-cloud-batch>=0.5.0; extra == "gcp-batch"
Requires-Dist: google-cloud-logging>=3.0.0; extra == "gcp-batch"
Requires-Dist: google-cloud-runtimeconfig>=0.33.2; extra == "gcp-batch"
Provides-Extra: kfp
Requires-Dist: kfp==1.6.2; extra == "kfp"
Provides-Extra: kubernetes
Requires-Dist: kubernetes>=11; extra == "kubernetes"
Provides-Extra: ray
Requires-Dist: ray>=1.12.1; extra == "ray"

[![PyPI](https://img.shields.io/pypi/v/torchx)](https://pypi.org/project/torchx/)
[![License](https://img.shields.io/badge/License-BSD%203--Clause-blue.svg)](https://github.com/pytorch/torchx/blob/main/LICENSE)
![Tests](https://github.com/pytorch/torchx/actions/workflows/python-unittests.yaml/badge.svg)
![Lint](https://github.com/pytorch/torchx/actions/workflows/lint.yaml/badge.svg)
[![codecov](https://codecov.io/gh/pytorch/torchx/branch/main/graph/badge.svg?token=ceHHIm0hXy)](https://codecov.io/gh/pytorch/torchx)


# 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!](https://github.com/pytorch/torchx/issues?q=is%3Aopen+is%3Aissue+label%3Ascheduler-request)

## Quickstart

See the [quickstart guide](https://pytorch.org/torchx/latest/quickstart.html).

## Documentation

* [Stable Documentation](https://pytorch.org/torchx/latest/)
* [Nightly Documentation](https://pytorch.org/torchx/main/)

## Requirements

torchx:

* python3 (3.8+)
* [PyTorch](https://pytorch.org/get-started/locally/)
* optional: [Docker](https://docs.docker.com/get-docker/) (needed for docker based schedulers)

Certain schedulers may require scheduler specific requirements. See installation
for info.

## Installation

### Stable

```bash
# 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

```bash
# install torchx sdk and CLI
pip install torchx-nightly[dev]
```

### Source

```bash
# 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](https://github.com/pytorch/torchx/blob/main/CONTRIBUTING.md) file.

## License

TorchX is BSD licensed, as found in the [LICENSE](https://github.com/pytorch/torchx/blob/main/LICENSE) file.
