Thinking Machines 在 Connectionism 博客发文,介绍 Inkling 2 Small 如何从预训练第一步到最后一次 RL rollout 都用原生 4 位(NVFP4)前向计算:反向用 MXFP8 加随机舍入,通信也做量化。官方称这是已知最大的端到端 4 位前向训练模型,相比全 MXFP8 训练,预训练墙钟时间快 8%、RL 快 20%。

核心要点速览

  • ✓NVFP4 只用于 MoE 专家层的前向计算,反向用 MXFP8 并配合重量化和随机舍入,以减少梯度偏差
  • ✓同一套数值配置从预训练沿用到 SFT 和 RL,训练结束交付量化权重时无需再做校准
  • ✓相比全 MXFP8:预训练墙钟提速 8%,RL 提速 20%;预训练计算效率约损失 2%,RL 奖励无可见差异
  • ✓专家并行通信成为新瓶颈,团队用 32 维随机 Hadamard 变换加块缩放格式量化 combine 通信
  • ✓官方称同规模模型数月前训练需数周,Inkling 2 Small 在 GB300 NVL72 上 68 小时完成预训练
Thinking Machines 公开 Inkling 2 Small 端到端 4 位训练方案:专家层前向全程 NVFP4,预训练提速 8%、RL 提速 20%,594B 模型 68 小时完成预训练
🖼️要闻配图
点击全屏高清查看
🧭

把技术选型换算成你的开发预算

40 款主流编程套餐横向比价,支持输入/输出 Token 真实账单模拟

🔬

深度技术解析与实战评估

行业背景与痛点

低精度训练能省显存、算得更快,但位数越少,舍入误差越大,误差会随层数和训练步数累积,影响学习。NVIDIA GB300 上 NVFP4 每个值约 4.5 位,矩阵乘算力约 15 PFLOPS,是 MXFP8(5)的 3 倍、BF16(2.5)的 6 倍。难点在于怎样拿到速度又不伤精度。

架构亮点与底层机制

据官方博客(作者 Jack Cook),Inkling 2 Small 只在 MoE 专家层前向用 NVFP4,并配专用内核;反向用 MXFP8,加重量化和随机舍入减少偏差;最后 16.6% 的层保持 MXFP8。前向提速后专家并行通信成了瓶颈,团队先做 32 维随机 Hadamard 变换再量化通信数据。自研 MoE 内核栈代号 BBBK,计算完全确定。

权威 Benchmark 与实测跑分对比

官方数据:相比全 MXFP8,预训练墙钟快 8%,RL 快 20%;预训练计算效率约损失 2%,RL 奖励没有可见差异。官方称数月前同规模模型训练要数周,Inkling 2 Small 在 GB300 NVL72 上 68 小时完成预训练。官方未公布与其他实验室方案的直接对比。

开发者实战落地与开箱指南

这套方案对自训 MoE 的团队最有参考价值:优先量化专家层前向,反向保守用 8 位。博客附可交互的 NVFP4 量化演示和完整公式。想直接用成果,可在 Hugging Face 下载 Inkling 2 Small 的 NVFP4 权重。团队表示正在研究反向也用 FP4,以及在 Vera Rubin 上的新配方。

实战指南准备好将技术转化为生产力?

即刻查看该技术对应模型与主流工具的实测差异,或一键测算团队 API 调用与 $20/月套餐盈亏平衡点。

接着读Thinking Machines 兑现承诺开源 Inkling 2 Small:594B 总参 / 17B 激活 MoE 权重上架 Hugging Face,Apache 2.0 可商用,原生 NVFP4 两张 B300 即可跑,SGLang、vLLM、Cloudflare 等首日支持
立即阅读→