I wrote a minimal PyTorch FSDP to understand how it works (~240 LOC) | Heykuki News