distributed-training-pytorch-fsdp — Claude Code skill: install
This skill guides PyTorch FSDP work, from parameter sharding to mixed precision and CPU offload. Use it when building or debugging distributed training code.
npx claude-code-templates@latest --skill ai-research/distributed-training-pytorch-fsdpWhat distributed-training-pytorch-fsdp does
The skill covers PyTorch FSDP concepts and implementation work, including parameter sharding, mixed-precision training, CPU offloading, and FSDP2. It is aimed at questions about the API, setup decisions, and debugging distributed training code.
Its reference material also includes torch.distributed and the Join context for workers that process different amounts of input. Joinable and JoinHook show how custom hooks can keep collective operations aligned when one process finishes early. The listed dependencies are torch>=2.0 and transformers, so a project using an older PyTorch release does not meet the stated requirement.
What it helps with
- configure parameter sharding with PyTorch FSDP
- choose a mixed-precision setup
- move a training design to FSDP2
- handle uneven input counts across workers
- debug code built on
torch.distributed
Example prompts for the agent
- Explain how to set up parameter sharding and mixed precision with PyTorch FSDP.
- Show how to use
Joinwhen workers receive different numbers of inputs. - Review this
torch.distributedtraining code and suggest a fix for the error.
How to install distributed-training-pytorch-fsdp
1claude-code-templates
npx claude-code-templates@latest --skill ai-research/distributed-training-pytorch-fsdp2Skills CLI (Vercel)
npx skills add davila7/claude-code-templates --skill distributed-training-pytorch-fsdp -a claude-code3Manually: put the files in place
~/.claude/skills/distributed-training-pytorch-fsdp/SKILL.mdCopy the whole skill folder; for one project use .claude/skills/distributed-training-pytorch-fsdp/
Manually: put the files in place
.agents/skills/distributed-training-pytorch-fsdp/SKILL.mdFor all projects use ~/.agents/skills/
Manually: put the files in place
.cursor/skills/distributed-training-pytorch-fsdp/SKILL.mdCursor also reads .agents/skills/ and .claude/skills/
1GitHub CLI
gh skill install davila7/claude-code-templates distributed-training-pytorch-fsdp2Manually: put the files in place
.github/skills/distributed-training-pytorch-fsdp/SKILL.mdPlatform CLI
gemini skills install https://github.com/davila7/claude-code-templates --path cli-tool/components/skills/ai-research/distributed-training-pytorch-fsdpManually: put the files in place
.devin/skills/distributed-training-pytorch-fsdp/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-pytorch-fsdp
How do I install distributed-training-pytorch-fsdp in Claude Code?
Run npx claude-code-templates@latest --skill ai-research/distributed-training-pytorch-fsdp.
What does distributed-training-pytorch-fsdp do?
It provides guidance for PyTorch FSDP, including parameter sharding, mixed precision, CPU offload, and FSDP2. It also covers API questions and debugging for distributed training.
What does distributed-training-pytorch-fsdp require?
The listed dependencies are torch>=2.0 and transformers. A PyTorch version below 2.0 does not satisfy the skill’s requirement.