Skip to main content
skill

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.

Install into Claude Code · claude-code-templates All methods and platforms ↓
npx claude-code-templates@latest --skill ai-research/distributed-training-deepspeed
Open on GitHub

What 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-deepspeed

2Skills CLI (Vercel)

npx skills add davila7/claude-code-templates --skill distributed-training-deepspeed -a claude-code

3Manually: put the files in place

~/.claude/skills/distributed-training-deepspeed/SKILL.md

Copy the whole skill folder; for one project use .claude/skills/distributed-training-deepspeed/

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.

Similar skills

All skills for AI and ML →

From the same set804

The whole set →
professional-communication This skill helps draft workplace emails, team updates, and meeting agendas. Use it when a technical issue needs a clear explanation for managers, teammates, or non-technical stakeholders. 104installs snowflake-semanticview Generates, alters, and validates Snowflake semantic views through Snowflake CLI. It enforces star schema modeling alongside automated CLI validation. 104installs django-pro Get Django guidance for project structure, ORM queries, APIs, tests, and deployment. Useful when building or maintaining apps with Django 5.x and its surrounding tools. 103installs edit-audit-log This hook records the path and timestamp for files changed with Edit. Use it to review file changes across Claude Code sessions. 103installs hypothesis-generation Formulate testable scientific hypotheses from observations and literature data. Helps researchers plan experimental designs and structure academic LaTeX reports. 101installs templates Discovers Railway marketplace templates and deploys pre-configured services. Useful when adding apps like Ghost, Strapi, or n8n to your project. 101installs deployment-procedures This skill guides production releases, post-deploy checks, and rollback choices. Use it when a project needs a platform-specific deployment plan. 101installs distributed-training-pytorch-fsdp This skill guides PyTorch FSDP work, from parameter sharding to mixed precision and CPU offload. Use it when building or debugging distributed training code. 101installs

Project sponsors

Companies that keep this analytics open
Gold sponsor

Impressions, clicks and code copies — open statistics. We keep the numbers public for the same reason NeBlask itself is public. Become a sponsor