Registry / ai-ml / torchx-nightly

torchx-nightly

library2026.6.19pypypi✓ verified 80d ago

TorchX is a Python SDK and set of components for building and launching distributed PyTorch applications. It provides a unified API to run jobs across different schedulers (local, Slurm, Kubernetes, AWS Batch, etc.). Version 2026.4.30 is a nightly release. Release cadence is nightly (torchx-nightly) vs stable (torchx) using semver.

ai-ml devops workflow

pip install torchx-nightly

github ↗package ↗homepage ↗docs ↗

INSTALL

IMPORT

SIG · TORCHX-NIGHTLY

torchx-nightly

ai-mlpythonv2026.6.19

Install

3.3s avg

Import

—

Disk

28MB

Pass rate

10/ 10

Env Coverage10 / 10

glibc

3.9–3.13

musl

3.9–3.13

Install & Compatibility

Where this runs

tested against v2026.6.19 · pip install

no network on importno background threads

Install × environment matrix

Each cell = how many times install + import succeeded across repeated harness runs. Partial = flaky.
glibc = Debian/Ubuntu slim · musl = Alpine Linux

musl

py 3.10–3.95 runs

installs and imports cleanly · install 0.0s · import 0.000s · 28.9MB

glibc

py 3.10–3.95 runs

installs and imports cleanly · install 3.3s · import 0.000s · 30MB

28MB installed

● package 28MB

Code

Verified usage

Verified import paths — ran on the pinned version, not inferred.

IMAGE

✓ from torchx import IMAGE

✗ from torchx import TorchX

util

✓ from torchx import util

version

✓ from torchx import version

Create a simple AppDef and schedule it using the local (current working directory) scheduler.

import os from torchx import TorchX from torchx.specs import AppDef, Role app = AppDef( name="test_job", roles=[ Role( name="worker", image="pytorch/pytorch:latest", entrypoint="python", args=["-c", "import torch; print('hello')"], num_replicas=2, ) ], ) scheduler = TorchX.scheduler("local_cwd") scheduler.schedule(app) print("Job scheduled successfully")

torchx --version

Debug

Known issues

breakingIn version 0.6.0, the `run` function moved from `torchx` to `torchx.runner`. Old code using `from torchx import run` will break.

fix

Use `from torchx.runner import run` instead.

affects: >=0.6.0

breakingIn version 0.5.0, scheduler factories were moved from `torchx.runtime` to `torchx.schedulers`. Imports from the old path will fail.

fix

Use `from torchx.schedulers import get_scheduler_factories`.

affects: >=0.5.0

gotchaThe nightly package `torchx-nightly` can conflict with the stable `torchx` package. Installing both can cause import errors.

fix

Use a virtual environment and install only one: either `torchx` (stable) or `torchx-nightly` (unstable).

affects: all

gotchaTorchX requires PyTorch >= 1.11. If not installed, you'll get ImportError on torchx installation.

fix

Install PyTorch separately via `pip install torch` (or the appropriate variant for your system).

affects: all

deprecatedThe `AppDef` constructor argument `dist_role` is deprecated as of 0.7.0. Use the new `Role` API instead.

fix

Define roles explicitly using `Role` objects and pass them to `AppDef(roles=[...])`.

affects: >=0.7.0

gotchaWhen using `local_cwd` scheduler, the working directory is the current directory. If job artifacts or data are relative, they may not be found.

fix

Use absolute paths or set `workspace` in the TorchX config.

affects: all

gotchaFor remote schedulers (Kubernetes, Slurm) you need additional dependencies like `torchx[kubernetes]`. Missing them results in ModuleNotFoundError.

fix

Install extra dependencies: `pip install torchx[kubernetes]` or `torchx[slurm]`, etc.

affects: all

Upgrade

Version history

2026.6.19latest on PyPI · released Jun 19, 2026

Audit

Dependencies

torchrequiredCore dependency for torchx components

Agent activity

61 hits · last 30 days

node

OpenAI (training)

Resources

githubhttps://github.com/meta-pytorch/torchx ↗

packagehttps://pypi.org/project/torchx-nightly/ ↗

homepagehttps://pytorch.org/torchx ↗

docshttps://meta-pytorch.org/torchx/ ↗