deepspeed

Implement and tune DeepSpeed training workflows with ZeRO stages, pipeline parallelism, mixed precision, and MoE on GPU clusters.

Updated Apr 3, 2026
One-click install
npx skills add https://github.com/handsomelong922/my-codex-skills --skill deepspeed-handsomelong922
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: deepspeed
Source: https://github.com/handsomelong922/my-codex-skills/tree/main/skills/deepspeed
Command: npx skills add https://github.com/handsomelong922/my-codex-skills --skill deepspeed-handsomelong922

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill includes references (resource) and assets (resource) components.

What problem does it solve?

DeepSpeed provides expert guidance and practical workflows to efficiently train and optimize extremely large AI models, helping teams overcome memory, compute, and infrastructure barriers through ZeRO optimization, pipeline parallelism, and advanced precision techniques.

Core Features & Use Cases

  • ZeRO optimization stages (1-3) to dramatically reduce memory footprint during training.
  • Pipeline parallelism and 3D parallelism for scaling across multiple GPUs and nodes.
  • Mixed precision training including FP16, BF16, and FP8, plus 1-bit Adam for lower communication.
  • Sparse attention and DeepSpeed-MoE support for large models; real-world guidance and best practices.
  • Use case examples: training megamodels like GPT/MT-NLG scale, BERT-like models with large sequences, and multi-GPU inference optimization.

Quick Start

Provide a starter plan to configure DeepSpeed ZeRO-enabled training with pipeline parallelism and mixed precision.

Frequently Asked Questions about deepspeed

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

FAQPage Schema
How do I configure ZeRO optimization stages for distributed training?

To configure ZeRO optimization for distributed training, you apply specific ZeRO stages (1-3) to partition optimizer states, gradients, and parameters, dramatically reducing the memory footprint across GPUs. The Skill provides actionable setup instructions and configuration templates to implement this memory optimization.

What is pipeline parallelism and when do I need it for large GPU clusters?

Pipeline parallelism splits model layers across multiple GPUs and nodes to scale training for extremely large AI models. You need it when a single GPU cannot hold the model, enabling 3D parallelism to overcome memory and compute barriers across large GPU clusters.

Can I use mixed precision training with FP16, BF16, and FP8 in DeepSpeed?

Yes, you can use mixed precision training with FP16, BF16, and FP8 in DeepSpeed to accelerate compute and reduce memory. The Skill offers expert guidance on configuring these precision techniques alongside 1-bit Adam for lower communication overhead.

How do I implement DeepSpeed-MoE and sparse attention for large models?

To implement DeepSpeed-MoE and sparse attention for large models, you apply Mixture of Experts and sparse attention techniques to scale model parameters efficiently. The Skill provides real-world guidance and best practices for deploying these features.

Does DeepSpeed support multi-GPU inference optimization for GPT-scale models?

Yes, DeepSpeed supports multi-GPU inference optimization for GPT and MT-NLG scale models. The Skill provides actionable configuration templates and references to official documentation to optimize inference workloads across distributed GPU environments.