ray-train

Orchestrates PyTorch, TensorFlow and HuggingFace distributed training across clusters with hyperparameter tuning and fault tolerance.

Updated Apr 30, 2026
One-click install
npx skills add https://github.com/Ced3-han/Harness-Settings --skill ray-train-ced3-han
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: ray-train
Source: https://github.com/Ced3-han/Harness-Settings/tree/main/skills/ray-train
Command: npx skills add https://github.com/Ced3-han/Harness-Settings --skill ray-train-ced3-han

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 solves the problem of scaling machine learning training across multiple machines, from a single GPU to thousands of nodes, with built-in hyperparameter tuning and fault tolerance.

Core Features & Use Cases

  • Distributed Training: Scales PyTorch, TensorFlow, and HuggingFace training from single GPU to multi-node clusters.
  • Hyperparameter Tuning: Integrates with Ray Tune for efficient hyperparameter optimization.
  • Fault Tolerance: Ensures training can recover from failures.
  • Elastic Scaling: Dynamically adds or removes nodes during training to optimize resource usage.
  • Use Case: Ideal for training massive models across multiple machines or running distributed hyperparameter sweeps.

Quick Start

Install Ray Train with pip install -U "ray[train]" and start your distributed training script.

Frequently Asked Questions about ray-train

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

FAQPage Schema
How do I scale PyTorch training across multiple machines?

To scale PyTorch training across multiple machines, you can orchestrate distributed training from a single GPU to multi-node clusters using Ray Train, which manages the underlying execution and scaling logic automatically.

Does distributed training with Ray support HuggingFace transformers?

Yes, distributed training with Ray supports HuggingFace transformers natively. It orchestrates training across clusters for PyTorch, TensorFlow, and HuggingFace models to handle large-scale workloads efficiently.

Can I run hyperparameter tuning alongside distributed training?

You can run hyperparameter tuning alongside distributed training by integrating Ray Tune, enabling efficient hyperparameter optimization while simultaneously executing your training scripts across allocated cluster resources.

What frameworks do I need to install for multi-node machine learning training?

To run multi-node machine learning training, you need to install the ray[train] package alongside specific machine learning libraries like torch and transformers, depending on the framework your models use.

How does fault tolerance work during distributed machine learning training?

Fault tolerance during distributed machine learning training ensures that your execution can automatically recover from unexpected node failures, preventing the loss of training progress across your cluster.

Can I dynamically add or remove compute nodes during model training?

You can dynamically add or remove compute nodes during model training through elastic scaling, allowing you to optimize resource usage by adjusting cluster capacity based on current workload demands.