Skip to main content

TorchX SDK and Components

Project description

PyPI License Tests Lint codecov

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.

Requirements

TorchX SDK (torchx):

  • python3 (3.8+)
  • torch

TorchX Kubeflow Pipelines Support (torchx-kfp):

  • torchx
  • kfp

Installation

Release

# 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

# install torchx sdk and CLI
pip install torchx-nightly

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

Quickstart

See the quickstart guide.

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 Distributions

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distribution

torchx_nightly-2021.12.26-py3-none-any.whl (155.7 kB view details)

Uploaded Python 3

File details

Details for the file torchx_nightly-2021.12.26-py3-none-any.whl.

File metadata

  • Download URL: torchx_nightly-2021.12.26-py3-none-any.whl
  • Upload date:
  • Size: 155.7 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.7.1 importlib_metadata/4.10.0 pkginfo/1.8.2 requests/2.26.0 requests-toolbelt/0.9.1 tqdm/4.62.3 CPython/3.8.12

File hashes

Hashes for torchx_nightly-2021.12.26-py3-none-any.whl
Algorithm Hash digest
SHA256 92a7bb5f8506cfb9011afc7e608b9d671454b424cb6f3b8aba233088e07cbd2e
MD5 f5c99cd89f9bce46245fb1dc92527258
BLAKE2b-256 e78f80b610d9609ff2bccbcb8969c7c2e9e7914c322fbf30759f00d1d6a70ef9

See more details on using hashes here.

Provenance

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page