Skip to main content
skill

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.

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

What 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 Join when workers receive different numbers of inputs.
  • Review this torch.distributed training 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-fsdp

2Skills CLI (Vercel)

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

3Manually: put the files in place

~/.claude/skills/distributed-training-pytorch-fsdp/SKILL.md

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

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.

Similar skills

All skills for AI and ML →

From the same set495

The whole set →

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