Read the Scaling PyTorch models on Cloud TPUs with FSDP blog post.