distributed-training-deepspeed — Claude Code skill: what it does
Guides distributed training setups with DeepSpeed, ZeRO memory offloading, and DeepNVMe I/O. Built for debugging ML pipelines and mixed-precision runs.
npx claude-code-templates@latest --skill ai-research/distributed-training-deepspeedWhat distributed-training-deepspeed does
This skill provides structured assistance for Microsoft DeepSpeed workflows. It covers ZeRO memory partitioning stages, pipeline parallelism, sparse attention, and mixed-precision formats including FP16, BF16, and FP8, along with 1-bit Adam optimization.
A central section explains the DeepNVMe subsystem. It explains how to instantiate aio_handle and gds_handle using builder classes, execute blocking writes, and coordinate non-blocking disk transfers via .wait() synchronization.
Using these workflows requires DeepSpeed 0.15.0 or newer alongside PyTorch, Accelerate, and Transformers. Without the host libaio-dev package, the core async_io operator will fail to build, which reveals itself as an error in ds_report.
What it helps with
- configure ZeRO optimization stages to reduce GPU memory footprint
- implement non-blocking tensor I/O with DeepNVMe handles
- diagnose missing C++ kernel extensions through ds_report
- set up FP8, BF16, or FP16 mixed precision for large runs
Example prompts for the agent
- Show me how to save host and CUDA tensors using aio_handle in DeepNVMe
- How do I resolve missing libaio-dev warnings shown by ds_report?
- Configure ZeRO Stage 3 optimization for my PyTorch training script
How to install distributed-training-deepspeed
1claude-code-templates
npx claude-code-templates@latest --skill ai-research/distributed-training-deepspeed2Skills CLI (Vercel)
npx skills add davila7/claude-code-templates --skill distributed-training-deepspeed -a claude-code3Manually: put the files in place
~/.claude/skills/distributed-training-deepspeed/SKILL.mdCopy the whole skill folder; for one project use .claude/skills/distributed-training-deepspeed/
Manually: put the files in place
.agents/skills/distributed-training-deepspeed/SKILL.mdFor all projects use ~/.agents/skills/
Manually: put the files in place
.cursor/skills/distributed-training-deepspeed/SKILL.mdCursor also reads .agents/skills/ and .claude/skills/
1GitHub CLI
gh skill install davila7/claude-code-templates distributed-training-deepspeed2Manually: put the files in place
.github/skills/distributed-training-deepspeed/SKILL.mdPlatform CLI
gemini skills install https://github.com/davila7/claude-code-templates --path cli-tool/components/skills/ai-research/distributed-training-deepspeedManually: put the files in place
.devin/skills/distributed-training-deepspeed/SKILL.mdThe legacy .windsurf/skills/ path still works
Claude.ai
Zip the folder and upload it in Settings → Capabilities → Skills
Step by step, with commands for each platform: How to install skills in Claude Code: five ways
Questions about distributed-training-deepspeed
How to install distributed-training-deepspeed in Claude Code?
Run npx claude-code-templates@latest --skill ai-research/distributed-training-deepspeed. You can also execute npx skills add davila7/claude-code-templates --skill distributed-training-deepspeed -a claude-code or place the definition in ~/.claude/skills/distributed-training-deepspeed/SKILL.md.
What does distributed-training-deepspeed do?
It assists developers with DeepSpeed APIs, ZeRO stages, pipeline parallelism, and high-performance tensor storage operations. It acts as an in-editor reference during ML model training setup.
What are the requirements for DeepNVMe support?
You need DeepSpeed 0.15.0 or newer and the Linux libaio-dev library. Run ds_report in your shell to verify that the async_io operator shows an OK status.