网易首页 > 网易号 > 正文 申请入驻

4192参数纯C语言GPT推理破千万tps,性能反超五倍参数模型

0
分享至

一个纯C语言编写、零依赖的微型GPT项目,最近在Apple M5 Pro上跑出了惊人的10168430 tok/sec(约1000万token每秒)推理速度。这个名为microGPT的项目,整个实现只有一个C文件,不依赖任何第三方库,仅靠libc标准库就能完成GPT的训练和推理。

项目作者在GitHub上开源了全部代码,支持macOS、Linux和Windows(MSYS2)三大平台,并在ARM64架构上利用NEON指令集、在x86-64上利用AVX2指令集做了底层优化。Makefile会自动检测宿主机架构并选择对应的编译参数,用户只需要执行make run就能跑起来。

训练速度:2万步只需几秒

microGPT是一个字符级别的transformer模型,实现了前向传播、反向传播、Adam优化器和采样功能。在包含约3.2万个名字的数据集上,它只需要几秒钟就能完成训练。从训练日志可以看到,模型在20000步训练中,损失从2.6036逐步下降到2.2201左右,训练过程相当稳定。

更让人意外的是,这个微型模型展现出了真正的泛化能力,而不是死记硬背。模型参数量只有4192个,在32033个名字中取20000个进行训练后,在训练集上的表现是2.2054 nats/字符,而在从未见过的12033个名字上,得分反而更好,达到2.2039 nats/字符。

小参数大能量:比五倍参数模型还强

作为对比,一个经过插值处理的trigram模型,参数量几乎是microGPT的五倍,但在同样的测试条件下,表现却不如这个4192参数的小家伙。这说明microGPT学到的不只是简单的统计规律,而是真正捕捉到了名字数据中的深层结构。

项目在工程实现上也有不少巧思。训练和推理使用了分离的前向传播路径:gpt_forward会存储激活值用于反向传播,而gpt_forward_infer则是专门优化的单token推理路径,两者的logits输出在fp32精度范围内完全一致。项目文档中专门有一份docs/PERFORMANCE.md,详细解释了这条推理路径的工作原理和性能瓶颈所在。

推理性能:从名字生成看实际效果

从实际生成效果来看,模型在训练后能生成kayley、maria、arana、shayan、jayden等听起来相当自然的人名。虽然模型很小,但生成的名字已经具备了英文名字的基本音韵特征,没有出现明显的乱码或拼写错误。

这个项目最大的价值在于,它把GPT的核心机制压缩到了极致——一个文件、零依赖、几秒钟训练、千万级token吞吐。对于想深入理解transformer原理的开发者来说,这几乎是最小可用的学习样本;对于需要在资源受限环境跑推理的场景,这种极简实现也提供了另一种思路。

代码已经开源,支持直接在任何语料上训练,每行一个样本即可。想试试的同学,克隆仓库后直接make run就能复现全部结果。

特别声明:以上内容(如有图片或视频亦包括在内)为自媒体平台“网易号”用户上传并发布,本平台仅提供信息存储服务。

Notice: The content above (including the pictures and videos if any) is uploaded and posted by a user of NetEase Hao, which is a social media platform and only provides information storage services.

相关推荐
热点推荐
砸神像,睡坟地,结阴婚,“反迷信”网红张树林,如今怎么样了?

砸神像,睡坟地,结阴婚,“反迷信”网红张树林,如今怎么样了?

皮皮电影
2026-08-23 10:20:22
中国3大通血管食物,洋葱排第3!第1名天天见,可惜很多人不知道

中国3大通血管食物,洋葱排第3!第1名天天见,可惜很多人不知道

江江食研社
2026-08-22 22:30:04
普京警告:乌克兰已“打开潘多拉魔盒”

普京警告:乌克兰已“打开潘多拉魔盒”

观察者网
2026-08-23 08:11:04
河北“甲醛泡白菜”正装车发往江苏!蘸一下,能保鲜两三天

河北“甲醛泡白菜”正装车发往江苏!蘸一下,能保鲜两三天

听心堂
2026-08-22 12:51:49
住破房、全身衣服不到20元,男子省吃俭用20年攒下近800万,跟陌生人炒股6天被骗光,骗子嘲讽:你是我有史以来最大的客户

住破房、全身衣服不到20元,男子省吃俭用20年攒下近800万,跟陌生人炒股6天被骗光,骗子嘲讽:你是我有史以来最大的客户

大风新闻
2026-08-23 10:07:12
“甲醛白菜”流向仍在追溯,三招教你识别!人民日报:性质恶劣,荒唐至极,眼下最紧迫的任务就是一追到底,及时拦截

“甲醛白菜”流向仍在追溯,三招教你识别!人民日报:性质恶劣,荒唐至极,眼下最紧迫的任务就是一追到底,及时拦截

都市快报橙柿互动
2026-08-22 23:37:34
勇敢的中国人民何惧白菜蘸甲醛

勇敢的中国人民何惧白菜蘸甲醛

阿亮评论
2026-08-22 19:17:42
箭体倒地、着陆腿压断、发动机压损,火箭回收还算成功吗?

箭体倒地、着陆腿压断、发动机压损,火箭回收还算成功吗?

巅峰高地
2026-08-23 10:23:40
殡葬师说漏嘴:人死后,无论火葬还是土葬,千万别买寿衣!

殡葬师说漏嘴:人死后,无论火葬还是土葬,千万别买寿衣!

小陆搞笑日常
2026-08-23 11:02:08
14岁王植兴,确认见义勇为

14岁王植兴,确认见义勇为

极目新闻
2026-08-22 20:48:03
恶劣!对手朝着倒地的维尼修斯又踢了一脚 穆帅暴怒:车轮战太过分

恶劣!对手朝着倒地的维尼修斯又踢了一脚 穆帅暴怒:车轮战太过分

风过乡
2026-08-23 10:48:07
王毅访问韩国,不到24小时,韩国宣布:拟对中国征收反倾销税

王毅访问韩国,不到24小时,韩国宣布:拟对中国征收反倾销税

小莜读史
2026-08-22 18:57:09
9月1日实施人民币新规,家里旧钞票,哪些能交易哪些不能

9月1日实施人民币新规,家里旧钞票,哪些能交易哪些不能

阿振观点
2026-08-23 06:04:38
重温解放日报评论:警惕宇树IPO资本泡沫!上市当天网下机构90%无限售梁文锋们可能已经套现数钱...

重温解放日报评论:警惕宇树IPO资本泡沫!上市当天网下机构90%无限售梁文锋们可能已经套现数钱...

新浪财经
2026-08-23 09:46:14
痛风糖果竟掺处方药废料,成本仅7元卖到598元,涉事企业被罚没款3.8亿余元,责任人被追究刑事责任,细节披露

痛风糖果竟掺处方药废料,成本仅7元卖到598元,涉事企业被罚没款3.8亿余元,责任人被追究刑事责任,细节披露

极目新闻
2026-08-23 09:50:50
窜台议员被拒绝入境,韩方向中国驻韩使馆递抗议信,中方懒得搭理

窜台议员被拒绝入境,韩方向中国驻韩使馆递抗议信,中方懒得搭理

听风行江湖
2026-08-23 08:38:11
马锦烜不幸离世

马锦烜不幸离世

政知新媒体
2026-08-23 10:46:35
许家印的无期判了,超1万亿供应商欠款,正在拖垮大量实体企业

许家印的无期判了,超1万亿供应商欠款,正在拖垮大量实体企业

王新喜
2026-08-21 09:06:35
恒大二号人物夏海钧,15年捞18.55亿逃美,全球禁令下他能撑多久

恒大二号人物夏海钧,15年捞18.55亿逃美,全球禁令下他能撑多久

晓岇就是我
2026-08-21 16:27:34
河北甲醛白菜后续!多部门介入,第三车被扣,甲醛成本曝光太意外

河北甲醛白菜后续!多部门介入,第三车被扣,甲醛成本曝光太意外

寒士之言本尊
2026-08-23 01:10:03
2026-08-23 15:20:49
报错免疫体
报错免疫体
一名在需求评审和数据异常中反复横跳的产品运营。
260文章数 61关注度
往期回顾 全部

科技要闻

“英伟达发通知:涨价15%以上”

头条要闻

当着韩国外长的面 王毅直言:美国对朝敌视政策该改了

头条要闻

当着韩国外长的面 王毅直言:美国对朝敌视政策该改了

体育要闻

字母+汤神,有没有搞头?

娱乐要闻

新加坡媒体挑衅《龙餐馆》,人民网直接下场

财经要闻

加拿大宣布“全额对等报复”美国新关税

汽车要闻

华为乾崑全栈赋能/更倾向城市出行 猛士X700车展首秀

态度原创

教育
数码
家居
健康
公开课

教育要闻

班主任告家长书:2026年中小学秋季开学温馨提示!(转给家长)

数码要闻

华为FreeBuds 7无线耳机外观曝光,取消家族式水滴造型设计

家居要闻

2026建博会(广州) 公装联探展交流活动

“矫正神器”真能治好脊柱侧弯吗?

公开课

李玫瑾:为什么性格比能力更重要?

无障碍浏览 进入关怀版