pytorch-fsdp2

Orchestra-Research/AI-Research-SKILLs/08-distributed-training/pytorch-fsdp2

12.1k 0TeXData Analysis
PlatformUniversal

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.

All Skills