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.