2025 Python Packaging Survey is now live!  Take the survey now

TorchX SDK and Components

Navigation

Verified details

These details have been verified by PyPI
Maintainers
Avatar for d4l3k from gravatar.com d4l3k Avatar for facebook from gravatar.com facebook Avatar for kiukchung from gravatar.com kiukchung Avatar for priyaramani from gravatar.com priyaramani

Unverified details

These details have not been verified by PyPI
Project links
Meta
  • License: BSD License (BSD-3)
  • Author: TorchX Devs
  • Tags pytorch , machine learning
  • Requires: Python >=3.7
Classifiers
Report project as malware

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.

Supported by

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