uv-ray-train

Orchestrates distributed ML training across multi-node clusters using Ray Train.

1|Updated Feb 6, 2026
One-click install
npx skills add https://github.com/uv-xiao/pkbllm --skill uv-ray-train
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: uv-ray-train
Source: https://github.com/uv-xiao/pkbllm/tree/main/knowledge/ML/distributed/ray-train
Command: npx skills add https://github.com/uv-xiao/pkbllm --skill uv-ray-train

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill requires ray[train], torch, transformers, and includes scripts (resource) and references (resource) components.

What problem does it solve?

This Skill addresses the challenge of scaling machine learning model training from a single machine to large, distributed clusters, simplifying the process of training massive models and performing hyperparameter tuning across multiple nodes.

Core Features & Use Cases

  • Distributed Training: Seamlessly scales PyTorch, TensorFlow, and HuggingFace models across multiple GPUs and nodes with minimal code changes.
  • Hyperparameter Tuning: Integrates with Ray Tune for efficient, distributed hyperparameter optimization.
  • Fault Tolerance & Checkpointing: Automatically handles worker failures and resumes training from saved checkpoints.
  • Use Case: Train a large language model on a cluster of 100 GPUs, or run a hyperparameter sweep for a complex deep learning model across 32 nodes.

Quick Start

Use the uv-ray-train skill to scale your PyTorch training script across 4 GPUs.

Frequently Asked Questions about uv-ray-train

High-intent search queries and answers about installing and using this skill.

FAQPage Schema
How do I scale PyTorch training across multiple GPUs and nodes?

Ray Train orchestrates distributed PyTorch training across multi-node clusters with minimal code changes, scaling models from a single machine to multiple GPUs and nodes seamlessly.

What is the best way to run distributed hyperparameter tuning with HuggingFace models?

Integrating Ray Tune with Ray Train enables distributed hyperparameter tuning for HuggingFace models, executing efficient hyperparameter optimization across multiple nodes simultaneously.

Does Ray Train support TensorFlow distributed training on multi-node clusters?

Yes, Ray Train supports TensorFlow distributed training on multi-node clusters, scaling models across multiple GPUs with fault tolerance, automatic checkpointing, and elastic scaling.

How does fault tolerance work during distributed machine learning training?

Fault tolerance in distributed ML training works by automatically handling worker failures and resuming training from saved checkpoints, ensuring progress is not lost during large-scale execution.

Can I use Ray Train for large language model training on 100 GPUs?

Yes, Ray Train supports large language model training on 100 GPUs, scaling from single-node to thousands-of-nodes deployments for massive model training and hyperparameter sweeps.

What are the limitations of scaling machine learning training with Ray Train?

Scaling ML training with Ray Train requires dependencies on PyTorch, TensorFlow, or HuggingFace frameworks, and needs adequate cluster resources to support elastic scaling and automatic checkpointing.