← All skills

pytorch-fsdp2

100 / 100 · Low risk

Fine to try.

By Orchestra-Research/AI-Research-SKILLs (GitHub) · coding · 14,447 stars · scanned at commit 773a529

Adds PyTorch FSDP2 (fully_shard) to training scripts with correct init, sharding, mixed precision/offload config, and distributed checkpointing. Use when models exceed single-GPU memory or when you need DTensor-based sharding with DeviceMesh.

Works with any agent that supports SKILL.md skills.

What this skill can do

We found nothing in the files we read that matches our rules. That does not prove the skill is safe: see the limits of static scanning.

Install

Paste this into your terminal. It downloads exactly the version we scanned, not whatever is newest.

if [ -e ~/.agents/skills/'pytorch-fsdp2' ] || [ -L ~/.agents/skills/'pytorch-fsdp2' ]; then echo 'A skill folder named pytorch-fsdp2 already exists — not replacing it' >&2; false; else d=$(mktemp -d) && git clone --quiet --filter=blob:none --no-checkout 'https://github.com/Orchestra-Research/AI-Research-SKILLs' "$d" && git -C "$d" checkout --quiet '773a52944ba4747a18bd4ae9ade53fff041adcbc' -- '08-distributed-training/pytorch-fsdp2' && mkdir -p ~/.agents/skills && cp -R "$d"/'08-distributed-training/pytorch-fsdp2' ~/.agents/skills/'pytorch-fsdp2' && rm -rf "$d"; fi

Codex reads ~/.agents/skills directly. For Claude Code, Cursor, Copilot, Gemini CLI, Antigravity or OpenCode, also link it. For Claude Code: ln -s ~/.agents/skills/'pytorch-fsdp2' ~/.claude/skills/'pytorch-fsdp2' (other agents: ~/.cursor/skills, ~/.copilot/skills, ~/.gemini/skills, ~/.gemini/config/skills, ~/.config/opencode/skills). Or install through the finder skill, which does both.

Badge for your README

The badge for Orchestra-Research/AI-Research-SKILLs shows the score of its lowest-scoring skill.

[![SkillTrust](https://zalavijaysinh.github.io/skilltrust/badges/Orchestra-Research/AI-Research-SKILLs.svg)](https://zalavijaysinh.github.io/skilltrust/repos/Orchestra-Research/AI-Research-SKILLs/)

Score history