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

소스 정보

저장소
GrayCodeAI/starling
최근 소스 활동
2026년 7월 13일 09:28
감지된 SKILL.md 언어
영어
스타
2
포크
0

설치 방법

기본적으로 소스를 먼저 확인하는 Prompt가 선택됩니다. 직접 명령으로 전환하거나 로컬 사본을 다운로드할 수도 있습니다.

소스 파일 검토

설치 여부를 결정하기 전에 SKILL.md와 SkillsMP에 표시된 보조 파일을 읽어 보세요.

SKILL.md 표시 중

SKILL.md
소스 지침 · 읽기 전용 미리보기
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)
GitHub에서 보기