{"categories":[{"label":"Distributed Computing","url":"https://skillfed.io/packages/category/system-distributed-computing/2"}],"enrichment":{"capability":"torchrunx distributes PyTorch training functions across multiple GPUs or machines using a functional API, eliminating the need for CLI-based launchers.","skillfed_tags":["pytorch","distributed-training","gpu-orchestration"],"use_cases":["Launch distributed training across multiple GPUs on a single machine without writing shell scripts","Coordinate multi-machine training with automatic SSH setup and result aggregation","Integrate distributed training into larger Python workflows that need to inspect or process per-rank outputs","Fine-tune large language models using Transformers or DeepSpeed with functional control","Run single-GPU training with automatic environment variable setup for compatibility"],"what_it_does":"torchrunx is a functional launcher for PyTorch distributed training that replaces CLI-based tools. Instead of shell commands, you write a single Python function and pass it to a Launcher object, which handles environment setup, rank assignment, and result collection across devices. It depends on torch, numpy, cloudpickle for serialization, and fabric for SSH communication.\n\nThe package is designed for multi-GPU and multi-machine setups, requiring Linux and SSH with a shared filesystem for multi-machine scenarios. It integrates with standard PyTorch distributed patterns (DistributedDataParallel, environment variables like RANK and LOCAL_RANK) and supports frameworks like Transformers, DeepSpeed, PyTorch Lightning, and Accelerate. Results from each rank are collected and returned to the caller, enabling complex workflows within a single script.","worth_installing":"Yes, with conditions. Install if you need functional, in-script control over distributed PyTorch training and are comfortable with GPLv3 copyleft terms. The low install friction and active maintenance are positive signals, but the young codebase and modest adoption mean it lacks the battle-testing of established alternatives. Best for teams already using GPLv3 or open-source projects; avoid if your codebase must remain proprietary."},"id":"torchrunx","links":{"html":"https://skillfed.io/packages/torchrunx","md":"https://skillfed.io/packages/torchrunx.md","pypi":"https://pypi.org/project/torchrunx/"},"maintenance":{"status":"active"},"meta":{"latest_release":"2026-08-04","license_spdx":null,"license_treatment":"copyleft","name":"torchrunx","python_support":"supports_current","summary":"Automatically initialize distributed PyTorch environments"},"popularity":{"monthly_downloads":161386,"position":10637,"tier":"top_15000"},"security":{"n_vulnerabilities":0},"version":"0.4.0"}
