home/categories/framework-internals/orchestra-research-ai-research-skills-08-distributed-training-pytorch-fsdp2-skill-md
framework-internalsdevelopment
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 need DTensor-based sharding with DeviceMesh.
maintainer
Orchestra-Research
अपडेट किया गया 1/30/2026
स्टार
6563
फोर्क
515
quick start
Installation and usage
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.
इंस्टॉलेशन
$ install --globalskills.sh
उपयोग
इंस्टॉल करने के बाद, आप टर्मिनल में यह कमांड चलाकर इस स्किल का उपयोग कर सकते हैं:
skills use pytorch-fsdp2