Every premium and members-only resource for $9.99/month, as a no-commitment subscription. See the plans

PalateKit

Sign in
Menu

Request a resource

Describe the resource you are missing in one sentence. We will email you when it is published.

0 / 280

A page, a video or an image online.

Unlock PalateKit

Unlock everything

Plan

Paying means accepting our terms of sale.

Secure payment by Stripe. Receipts and invoices sent by email.

Welcome back

Sign in to install, download or save this resource. A free account is enough for free resources.

Training with PyTorch FSDP v1.0.0

Use this resource to train a model with PyTorch FSDP

New

  • 427 downloads
  • 101 likes
  • 228 saves

Free

Install with the CLI

In a terminal, from the folder you want:

npx --yes palatekit@latest install pytorch-fsdp-sharded-training

With no option, the skill goes to ~/.agents/skills/; add --agent claude-code (or codex, cursor…) to target an agent.

First time: connect the CLI with

npx --yes palatekit@latest login

The agent helps you set up FSDP: parameter sharding, mixed precision, CPU offloading and moving to FSDP2. You get setup code and common patterns drawn from the reference documentation.

Mostly picked by

Updated November 2, 2026

Card for the “Training with PyTorch FSDP” skill

Details

Works in
Install guides
Generates with
Your agent's model
Licence
MIT

Similar resources

View all

What you get

  • SKILL.md: quick reference, common patterns and pointers to the documentation
  • references/index.md: index of FSDP topics
  • references/other.md: additional FSDP documentation notes

Good fit for

  • Your model does not fit in one GPU's memory even with DDP
  • You are moving from DDP to FSDP2 and want the right settings

Not for

    Tags

    • fsdp
    • pytorch
    • sharding
    • mixed precision

    About

    You say

    “Wrap my model with FSDP2 in BF16, with CPU offloading of the parameters.”

    Card for the “Training with PyTorch FSDP” skill

    Skill published by davila7 under the MIT license, delivered in its original version.

    Show more

    How to use it

    1. Install the skill in your agent's skills folder
    2. Ask your agent to train a model with PyTorch FSDP
    3. Review and adjust the result

    Content preview

    Files

    • LICENSE1.1 KB
    • NOTICE-PALATEKIT.md0.5 KB
    • references/index.md0.1 KB
    • references/other.md333.4 KB
    • SKILL.md156.6 KB

    Origin and licence

    Listed unmodified

    Author
    davila7
    Source
    github.com/davila7/claude-code-templates/tree/f57964ebda91cdfb31755661ecea9395f30020f1/cli-tool/components/skills/ai-research/distributed-training-pytorch-fsdp
    Licence
    MIT

    Notice

    « distributed-training-pytorch-fsdp » (davila7, MIT) livré dans sa version d'origine : fiche en français et en anglais par PalateKit.

    User reviews

    –

    No ratings yet

    Loading reviews…

    Delete your review?

    Your rating and comment will be removed from the page.