Metadata-Version: 2.1
Name: torchx
Version: 0.1.0
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
Platform: UNKNOWN
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: pyyaml
Requires-Dist: importlib-metadata ; python_version < "3.8"
Provides-Extra: dev
Requires-Dist: boto3 (>=1.11.11) ; extra == 'dev'
Requires-Dist: moto (>=2.0.6) ; extra == 'dev'
Requires-Dist: kfp (==1.6.2) ; extra == 'dev'
Requires-Dist: pyre-extensions (>=0.0.21) ; extra == 'dev'
Requires-Dist: black (>=21.5b1) ; extra == 'dev'
Requires-Dist: usort (==0.6.4) ; extra == 'dev'
Requires-Dist: pytorch-lightning (>=1.4.9) ; extra == 'dev'
Requires-Dist: torch (>=1.10.0) ; extra == 'dev'
Requires-Dist: torchvision (>=0.11.1) ; extra == 'dev'
Requires-Dist: torchtext (>=0.11.0) ; extra == 'dev'
Requires-Dist: classy-vision (>=0.6.0) ; extra == 'dev'
Requires-Dist: flake8 (==3.9.0) ; extra == 'dev'
Requires-Dist: ts (>=0.5.1) ; extra == 'dev'
Requires-Dist: torchserve (>=0.4.0) ; extra == 'dev'
Requires-Dist: captum (>=0.3.1) ; extra == 'dev'
Requires-Dist: importlib-metadata ; extra == 'dev'
Requires-Dist: ax-platform[mysql] (>=0.2.2) ; extra == 'dev'
Requires-Dist: fsspec[s3] (>=2021.09.0) ; extra == 'dev'
Requires-Dist: torch-model-archiver (>=0.4.2) ; extra == 'dev'
Provides-Extra: kfp
Requires-Dist: kfp (==1.6.2) ; extra == 'kfp'
Provides-Extra: kubernetes
Requires-Dist: kubernetes (>=11) ; extra == 'kubernetes'

[![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)](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 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](https://pytorch.org/torchx).


## Requirements
TorchX SDK (torchx):
* python3 (3.8+)
* torch

TorchX Kubeflow Pipelines Support (torchx-kfp):
* torchx
* kfp

## Installation

### Release

```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]"
```

### Nightly

```bash
# install torchx sdk and CLI
pip install torchx-nightly
```

### 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[kubernetes]
```

## Quickstart

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

## Contributing

We welcome PRs! See the [CONTRIBUTING](CONTRIBUTING.md) file.

## License

TorchX is BSD licensed, as found in the [LICENSE](LICENSE) file.


