Skip to content

[Feat] [FSDP2] [SFT] [ExpertParallel] [DeepSeekV4] fsdp2 dpsk v4 flash 0731 support on only 1Node with 8*H200 - #10197

Open
mugglewei97 wants to merge 2 commits into
modelscope:mainfrom
mugglewei97:feat/fsdp2-dpsk-v4-flash
Open

mugglewei97 wants to merge 2 commits into
modelscope:mainfrom
mugglewei97:feat/fsdp2-dpsk-v4-flash

Conversation

@mugglewei97

Copy link
Copy Markdown

PR type

New Feature
More Models or Datasets Support

PR information

为混合专家(MoE)模型(如 DeepSeek-V4、Mixtral、Qwen-MoE
等)增加了基于 FSDP2 后端、具备普适性的专家并行EP功能,通过A2A通信
使得仅在1台Hopper上即可实现full sft训练

Experiment results

Training script which has been actual tested is added
Paste your experiment result here(if needed).

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant