Skip to main content

research-pytorch-fsdp2

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...

Quellinformationen

Repository
GrayCodeAI/starling
Letzte Quellaktivität
13. Juli 2026 um 09:28
Erkannte Sprache von SKILL.md
Englisch
Sterne
2
Forks
0

Installationsoptionen

Standardmäßig ist der Prompt ausgewählt, der zuerst die Quelle prüft. Sie können zu einem direkten Befehl wechseln oder eine lokale Kopie herunterladen.

Quelldateien prüfen

Lesen Sie SKILL.md und alle von SkillsMP angezeigten Begleitdateien, bevor Sie sich für eine Installation entscheiden.

SKILL.md wird angezeigt

SKILL.md
Quellanweisungen · Schreibgeschützte Vorschau
name
research-pytorch-fsdp2
description
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...
license
MIT
tags
["pytorch","fsdp2","fully-sharded-data-parallel","distributed-training","dtensor"]
version
1.0.0
author
Orchestra Research
dependencies
["torch"]
## References - `references/pytorch_fsdp2_tutorial.md` - `references/pytorch_fully_shard_api.md` - `references/pytorch_ddp_notes.md` - `references/pytorch_fsdp1_api.md` - `references/pytorch_device_mesh_tutorial.md` - `references/pytorch_tp_tutorial.md` - `references/pytorch_dcp_overview.md` - `references/pytorch_dcp_recipe.md` - `references/pytorch_dcp_async_recipe.md` - `references/pytorch_examples_fsdp2.md` - `references/torchtitan_fsdp_notes.md` (optional, production notes) - `references/ray_train_fsdp2_example.md` (optional, integration example)
Auf GitHub ansehen