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 条相关