QwenGPU:在大模型内部模拟 GPU 训练神经网络
开发者 zeeshanp 展示 QwenGPU 项目系列实验:在 Qwen3.5-2B 模型内部以分布式数据并行方式训练 MLP,40 步将 XOR 任务准确率从 53% 提升到 98.8%;再按 Megatron 式张量并行在 4 个「模型 GPU」上跑 MNIST,达到 86.9% 准确率;还训练微型 latent diffusion 模型生成手写数字,76% 可被分类器识别。验证环节显示各集合通信操作与真实 NCCL 结果逐位一致。
2026-11-03 ~ 2026-11-03 · 4 条相关
- 在 LLM 内部模拟 GPU 训练神经网络:XOR 任务 40 步从 53% 到 98.8% — zeeshanp_ · 2026-11-03
- 4 个 QwenGPU 集合通信结果与真实 NCCL 逐位一致 — zeeshanp_ · 2026-11-03
- QwenGPU 第二弹:Megatron 式张量并行在模型内部跑 MNIST,准确率 86.9% — zeeshanp_ · 2026-11-03
- 模型内部训练微型扩散模型生成手写数字,76% 可被分类器识别 — zeeshanp_ · 2026-11-03