Skip to main content
miles-rl-training Provides guidance for enterprise-grade RL training using miles, a production-ready fork of slime. Use when training large MoE models with FP8/INT4, needing train-inference alignment, or requiring speculative RL for maximum throughput.
跳到安装 Skills Marketplace 发现并探索由社区构建的 Agent Skills
用 Codex 或 Claude 帮你安装 复制这段 Prompt,粘贴到 Codex、Claude 或其他助手里,让它检查 Skill 页面并帮你完成安装。
直接命令不会经过审查 Prompt;运行前请先检查来源。
npx skills add https://github.com/davila7/claude-code-templates --skill miles-rl-training命令会保持在同一行。复制前请横向滚动并检查完整内容。
想先保存到本地?可下载 SkillsMP 当前能够提供的文件。
下载 Zip 下载中... 同仓库更多 Skills Build AI agents that interact with computers like humans do - viewing screens, moving cursors, clicking buttons, and typing text. Covers Anthropic's Computer Use, OpenAI's Operator/CUA, and open-source alternatives. Critical focus on sandboxing, security, and handling the unique challenges of vision-based control. Use when: computer use, desktop automation agent, screen control AI, vision-based agent, GUI automation.
Expert in building products that wrap AI APIs (OpenAI, Anthropic, etc.) into focused tools people will pay for. Not just 'ChatGPT but different' - products that solve specific problems with AI. Covers prompt engineering for products, cost management, rate limiting, and building defensible AI businesses. Use when: AI wrapper, GPT product, AI tool, wrap AI, AI SaaS.
github-workflow-automation Automate GitHub workflows with AI assistance. Includes PR reviews, issue triage, CI/CD integration, and Git operations. Use when automating GitHub workflows, setting up PR review automation, creating GitHub Actions, or triaging issues.
name miles-rl-training description Provides guidance for enterprise-grade RL training using miles, a production-ready fork of slime. Use when training large MoE models with FP8/INT4, needing train-inference alignment, or requiring speculative RL for maximum throughput. version 1.0.0 author Orchestra Research license MIT tags ["Reinforcement Learning","MoE","FP8","INT4","Enterprise","SGLang","Megatron-LM"] dependencies ["sglang-router>=0.2.3","ray","torch>=2.0.0","transformers>=4.40.0"]
miles: Enterprise-Grade RL for Large-Scale Model Training
miles is a high-performance, enterprise-ready RL framework optimized for large-scale model post-training. Built as a production fork of slime, it addresses critical challenges in MoE training stability, low-precision training, and train-inference alignment.
When to Use miles Choose miles when you need:
Training 1TB+ MoE models (DeepSeek V3, Qwen3-MoE)
FP8 or INT4 quantization-aware training
Bit-wise identical train-inference alignment
Speculative RL for maximum throughput
Production stability with enterprise support
Consider alternatives when:
You want the research-grade original → use slime
You need flexible backend swapping → use verl
You want PyTorch-native abstractions → use torchforge
Key Features
Low-Precision Training
Unified FP8 : End-to-end FP8 for both inference and training
INT4 QAT : 1TB models on single-machine VRAM (H200)
Rollout Routing Replay (R3) : Bit-wise expert alignment for MoE
Performance Optimizations
Speculative RL : 25%+ rollout speedup with online SFT draft models
Zero-Copy Weight Sync : CUDA IPC zero-copy mapping
Partial Rollout : Recycle half-finished trajectories
Train-Inference Alignment
TIS/MIS : Truncated/Masked Importance Sampling for off-policy correction
Kernel-level optimization : FlashAttention-3, DeepGEMM integration
Installation
docker pull radixark/miles:latest
docker run --rm --gpus all --ipc=host --shm-size=16g \
-it radixark/miles:latest /bin/bash
git clone https://github.com/radixark/miles.git
cd miles
pip install -r requirements.txt
pip install -e .
Quick Start miles inherits slime's configuration system. Basic training:
python train.py \
--advantage-estimator grpo \
--model-name qwen3-30b-a3b \
--hf-checkpoint /path/to/qwen3-30b-a3b-hf \
--rollout-batch-size 512 \
--n-samples-per-prompt 8
Workflow 1: Large MoE Training Use this workflow for training large MoE models like DeepSeek V3 or Qwen3-MoE.
Prerequisites Checklist
Step 1: Environment Setup
export NVTE_FP8_BLOCK_SCALING_FP32_SCALES=1
export CUDA_DEVICE_MAX_CONNECTIONS=1
Step 2: Configure Training python train.py \
--actor-num-gpus-per-node 8 \
--rollout-num-gpus 8 \
--hf-checkpoint /path/to/deepseek-v3 \
--advantage-estimator grpo \
--tensor-model-parallel-size 8 \
--expert-model-parallel-size 4 \
--prompt-data /path/to/data.jsonl \
--num-rollout 3000
Verification Checklist
Workflow 2: Speculative RL Training Use this workflow for maximum rollout throughput with EAGLE speculative decoding.
How Speculative RL Works
Small draft model generates candidate tokens
Target model verifies in parallel
Draft model updated via online SFT to track policy
Step 1: Enable Speculative Decoding miles supports EAGLE speculative decoding via SGLang:
python train.py \
--actor-num-gpus-per-node 8 \
--hf-checkpoint /path/to/target-model \
--sglang-speculative-algorithm EAGLE \
--sglang-speculative-num-steps 3 \
--sglang-speculative-eagle-topk 1 \
--sglang-speculative-num-draft-tokens 4 \
--sglang-speculative-draft-model-path /path/to/draft-model \
--advantage-estimator grpo \
--prompt-data /path/to/data.jsonl
Step 2: Enable Online MTP Training (Optional) For online SFT of draft model during training:
--mtp-num-layers 1 \
--enable-mtp-training \
--mtp-loss-scaling-factor 0.2
Note : Online MTP training requires a torch dist checkpoint with MTP weights. Add --mtp-num-layers 1 during checkpoint conversion from HuggingFace.
Expected Speedup
Standard rollout : Baseline
Speculative RL : 25-40% faster rollout
With partial rollout : Additional 10-15% throughput
Configuration Reference
Cluster Resources (from slime) --actor-num-nodes 1
--actor-num-gpus-per-node 8
--rollout-num-gpus 8
--rollout-num-gpus-per-engine 2
--colocate
Megatron Parallelism (from slime) --tensor-model-parallel-size 8
--pipeline-model-parallel-size 2
--expert-model-parallel-size 4
Speculative Decoding (miles-specific) --sglang-speculative-algorithm EAGLE
--sglang-speculative-num-steps 3
--sglang-speculative-eagle-topk 1
--sglang-speculative-num-draft-tokens 4
--sglang-enable-draft-weights-cpu-backup
--sglang-speculative-draft-model-path /your/draft/model/path
Online MTP Training (miles-specific) --mtp-num-layers 1
--enable-mtp-training
--mtp-loss-scaling-factor 0.2
Key Features (Conceptual) The following features are documented in miles but specific CLI flags may vary. Consult the miles repository for latest configuration.
Unified FP8 Pipeline End-to-end FP8 sampling and training that eliminates quantization-induced discrepancy causing RL collapse in MoE models.
Rollout Routing Replay (R3) Records expert routing decisions during SGLang inference and replays them during Megatron training for bit-wise expert alignment.
During SGLang inference, expert routing decisions are recorded
Routing decisions stored in sample.rollout_routed_experts
During Megatron training, routing is replayed instead of recomputed
Ensures identical expert selection between train and inference
INT4 Quantization-Aware Training Enables single-machine deployment of 1TB+ models (e.g., on H200).
Memory Savings with INT4 :
Model Size BF16 VRAM INT4 VRAM Reduction 70B 140GB 45GB 3.1x 235B 470GB 150GB 3.1x 671B 1.3TB 420GB 3.1x
Train-Inference Alignment miles achieves "exactly 0 KL divergence" between training and inference through:
Flash Attention 3
DeepGEMM
Batch-invariant kernels from Thinking Machines Lab
torch.compile integration
Sample Data Structure miles uses the same Sample dataclass as slime with the rollout_routed_experts field for MoE routing replay:
@dataclass
class Sample :
prompt: str | list [dict ]
tokens: list [int ]
response: str
reward: float | dict
loss_mask: list [int ]
status: Status
metadata: dict
rollout_log_probs: list [float ]
rollout_routed_experts: list [list [int ]]
Common Issues and Solutions
Issue: FP8 Training Collapse Symptoms : Loss explodes, NaN values
Use block scaling: export NVTE_FP8_BLOCK_SCALING_FP32_SCALES=1
Reduce learning rate: --lr 5e-7
Ensure MoE routing is consistent between train/inference
Issue: Speculative Draft Drift Symptoms : Low acceptance rate over time
Enable online MTP training to keep draft model aligned
Reduce speculative steps: --sglang-speculative-num-steps 2
Use CPU backup: --sglang-enable-draft-weights-cpu-backup
Issue: Train-Inference Mismatch Symptoms : Policy divergence, reward collapse
Use TIS for off-policy correction: --use-tis --tis-threshold 0.9
Verify log probs match between SGLang and Megatron
Enable R3 for MoE models
Supported Models Family Models MoE Support DeepSeek R1, V3, V3.2 Full Qwen 2, 2.5, 3 (including MoE) Full Llama 3, 3.1, 3.3, 4 Dense only Gemma 2, 3, 3N Dense only GLM 4.5, 4.6, 4.7 Dense only MiniMax M2, M2.1 Full
Resources