前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >00后华裔小哥哈佛辍学组团挑战英伟达,史上最快AI芯片Sohu推理性能超H100二十倍!

00后华裔小哥哈佛辍学组团挑战英伟达,史上最快AI芯片Sohu推理性能超H100二十倍!

作者头像
新智元
发布2024-06-27 19:37:27
1350
发布2024-06-27 19:37:27
举报
文章被收录于专栏:新智元新智元
新智元报道

编辑:编辑部

【新智元导读】史上最快Transformer芯片诞生了!用Sohu跑Llama 70B,推理性能已超B200十倍,超H100二十倍!刚刚,几位00后小哥从哈佛辍学后成立的公司Etached,宣布再融资1.2亿美元。

有史以来最快的Transformer芯片,刚刚诞生了?

去年21岁哈佛辍学华裔小哥的神级创业故事,进化到了新的阶段。

这一次,号称专为LLM推理加速打造的Sohu芯片,已经成超越Groq,成为最强AI芯片。

如今的Sohu,在跑Llama 70B时每秒可生成高达500000个token的输出。

换句话说就是:1张Sohu≈20张H100≈10张B200。

在去年12月,Etced就曾火过一阵,当时两位哈佛辍学小哥的2人公司,估值已经高达3400万刀。当时Sohu号称LLM推理性能达到H100的10倍,单价吞吐量更是高达140倍。

而就在刚刚,Etced又宣布了炸裂的新消息:已筹到1.2亿美元。

就在上周,英伟达的市值达到3.3万亿美元,超过微软成为全球第一。而Etched相信,自己的35人团队能够击败英伟达。

他们的豪放预言,得到了硅谷真金白银的支持。投资名单中,是一众豪华的机构和硅谷大佬,包括Peter Thiel。

Primary Venture Partners和Positive Sum Ventures领投了这轮融资,机构投资者包括 Hummingbird、Fundomo、Fontinalis、Lightscape、Earthshot、Two Sigma Ventures和 Skybox Data Centers。 天使投资者包括Peter Thiel、Stanley Druckenmiller、David Siegel、Balaji Srinivasan、Amjad Masad、Kyle Vogt、Kevin Hartz、Jason Warner、Thomas Dohmke、Bryan Johnson、Mike Novogratz、Immad Akhund、Jawed Karim和Charlie Cheeve。

Etched创始人之一Gavin Uberti表示,「我们正在押注人工智能领域最大的赌注——一种只能运行Transformer模型的芯片,但其运行速度比GPU快几个数量级。也许注意力确实是你所需要的全部...」

Pika研究人员表示,每秒50万token简直太疯狂了。你可以在2秒内写出「圣经」,或者一年内生成15万亿token。

Sohu大约每秒能输出21个人一天说的话,简直太不真实了。

Thiel Fellowship的主任Alex Handy,对几位年轻人表示了高度的赞许:「投资Etched是对 AI 价值的战略性押注。他们的芯片解决了竞争对手不敢面对的可扩展性问题,挑战了同行中普遍存在的停滞状态。」

在他看来,正是因为几位小哥做了底层的工作,硅谷的程序员们才能继续安心编程,不必担心正在开发技术的底层出问题。

两年前开始的豪赌

这个传奇的故事,始于两年前。

2022年,几位创始人就打赌:Transformer将占领世界。

为此,他们花了两年时间打造Sohu,这是世界上第一个用于Transformer的专用芯片(ASIC)。

将Transformer架构烧录到Sohu芯片后,它无法运行大多数的传统AI模型:比如DLRMs、AlphaFold 2,或Stable Diffusion 2;也无法运行CNNs、RNNs或LSTMs。

但对于Transformer来说,Sohu确实是有史以来最快的芯片。但它还可以更快。

对于Llama 70B,Sohu的吞吐量每秒超过了50万个token,因此它可以构建在GPU上根本不可能实现的产品。

而且,它甚至比英伟达的下一代Blackwell(B200)更快、更便宜!

如今最先进的AI模型,无一不是基于Transformer的,比如ChatGPT、Sora、Gmini、Stable Diffusion 3等等。

如果Transformer被SSM、RWKV或任何新架构取代,那Sohu将毫无用处。

但是,创始人小哥表示,如果他们押对了,Sohu将彻底改变世界!

这,就是一场豪赌。

Scale是「超级智能」所需的全部?

五年内,AI模型在大多数标准化测试中,超越了人类。

这是怎么做到的?

因为Meta用于训练Llama 400B(2024年SOTA模型)的计算量,比OpenAI训练GPT-2(2019年SOTA模型)多5万倍。

OpenAI用Scaling Law预测了GPT-4性能,并预计GPT-5+同样适用

奥特曼曾说过,「扩展参数规模确实非常重要。当我们在太阳周围建造了一个戴森球(Dyson Sphere)之后,我们才可以考虑讨论是否应该停止这么做,但在那之前不应该停下来」。

也就是说,通过向LLM提供更多的算力和更好的数据,才使得AI变得更加智能。

不得不承认,参数规模扩展(Scale)将会是几十年来唯一持续有效的方法。

每个大型AI公司(谷歌、OpenAI/微软、Anthropic/亚马逊等)都在未来几年投入超1000亿美元来继续扩大LLM规模。

然而,再扩展1000倍必定是昂贵的,形象地解释,下一代数据中心的成本将超过一个小国的GDP。

按照目前的发展速度,我们的硬件、成本、财力根本无法跟得上。

GPU撞墙了

圣克拉拉,英伟达总部所在地,不愿意让人知道的小秘密是——

GPU性能其实没有变得更好,只是变得更大了。 过去四年里,芯片单位面积的计算能力(每平方毫米的TFLOPS)几乎没有提升。

就比如,英伟达的B200、AMD的MI300、英特尔的Gaudi 3,以及亚马逊的Trainium2都将2个芯片集成到一张卡上,以使其性能翻倍。

如下曲线图中,也可以看出,从2022年-2025年,AI芯片并没有真正变得更好只是变得更大了。

在过去四年中,计算密度(TFLOPS/mm^2)仅提高了约15%。

而现在,随着摩尔定律的放缓,提高芯片性能的唯一途径,便是让其走向「专用」化。

专用芯片,势在必行

在Transformer占领世界之前,许多公司构建了灵活的AI芯片和GPU来处理数百种不同的机器学习架构。

举几个例子:

却从来没有人制造,针对特定算法的AI芯片(ASIC)。

因为,全新的芯片项目需要花费5000万-1亿美元,甚至要用许多年的时间才能投入生产。

当Etched开始着手这一项目时,根本就不存在这样的市场。

突然间,这种情况一下子发生了转变。

ChatGPT出现之前,Transformer的推理市场规模大约是5000万美元。

而现在,已经达到数十亿美元。所有的科技巨头,初创公司都在使用Transformer模型。

大模型架构,都在走向融合趋同。自从GPT-2诞生以来,最先进的模型架构几乎保持一致!

OpenAI GPT系列、谷歌PaLM、Meta Llama,甚至特斯拉的全自动驾驶系统,都采用了Transformer架构。

当模型的训练成本超过10亿美元,推理成本超过100亿美元时,专用芯片的出现是必然的。

在这种巨大参数规模之下,即使只有1%的性能提升,也足以证明5000万-1亿美元的定制芯片项目是值得的。

实际上,ASIC的速度要比GPU快很多个数量级。

Transformer拥有巨大的护城河

Etched.ai的几位创始人表示,他们相信硬件彩票——能在硬件上运行得最快、最便宜的模型,就是获胜的那一个。

而Transformer拥有巨大的护城河,足以在替代方案成熟之前主导各大AI计算市场。

理由如下——

1. Transformer正在为每一个大型AI产品提供支持,无论是Agent、搜索还是聊天。

为了优化GPU去适应Transformer,AI实验室已经在研发上投入了数亿美元。

无论是当前还是下一代SOTA模型,都是基于Transformer的。

2. 随着未来几年模型训练的规模从10亿美元扩大到100亿美元,再到1000亿美元,测试新架构的风险也急剧上升。

与其重新测试Scalling law和性能,不如把时间花在基于Transformer的功能开发上,比如多token预测。

3. 当前的软件栈,已为Transformer进行了优化。每个流行的库(TensorRT-LLM、vLLM、Huggingface TGI等),都有在GPU上运行Transformer模型的专用内核。

许多建立在Transformer之上的功能,比如推测解码、树搜索等,在替代方案中都很难得到支持。

Sohu可以通过树搜索更好地编码,能够并行比较数百个响应

4. 未来的硬件栈,也将为Transformer进行优化。比如英伟达的GB200,对Transformer Engine就有特殊的支持。

随着像Sohu这样的ASIC进入市场,将会带来一种不可逆的趋势。

也就是说,作为「Transformer Killer」的模型需要在GPU上,运行得比Sohu上的Transformer更快。

而但凡出现这种情况,创始人表示,他们也会为此构建一款全新的ASIC!

Sohu支持多重推测解码,可以实时生成新内容

Sohu来了!

Sohu是世界上第一个Transformer ASIC。

通过专门化,Sohu获得了前所未有的性能。一台8xSohu服务器每秒可以处理超过50万个Llama 70B token,等效于160块H100 GPU。

Sohu仅支持Transformer推理,并支持当今所有的模型(Google、Meta、Microsoft、OpenAI、Anthropic等),以及处理未来模型的调整。

无论是Meta的Llama、谷歌的Gemini、OpenAI的GPT、Anthropic的Claude、还是Stability AI的Stable Diffusion 3等等,都可以。

由于Sohu只能运行一种算法,因此绝大多数的控制流逻辑可以被移除,从而允许其拥有更多的数学模块。

结果就是,Sohu的FLOPS利用率高达90%以上;相比之下,使用TRT-LLM的GPU仅为为30%左右。

如何实现比GPU更多的FLOPS?

目前最先进的算力——英伟达H200,在没有稀疏处理的情况下具有989 TFLOPS的FP16/BF16算力。(超过了Google的新Trillium芯片之一)

而2025年推出的GB200,将会在算力上增加25%(每个芯片1250 TFLOPS)。

由于GPU的大部分面积都用于可编程性,因此专注于Transformer可以容纳更多的算力。

从基本原理上来看,这一点可以很容易被证明:

制造一个FP16/BF16/FP8乘法加法电路(所有矩阵数学的基础构件)需要10,000个晶体管。H100 SXM有528个张量核心,每个核心有4×8×16个FMA电路。通过计算可以得到,H100有27亿个晶体管专用于张量核心。

实际上,H100拥有800亿个晶体管!这意味着在H100 GPU上的晶体管中,仅有3.3%用于矩阵乘法!

但问题是,如果想要为各种模型(CNN、LSTM、SSM等)都提供支持,就不得不采取这样的设计。

这时,如果选择只运行Transformer,就可以在芯片上容纳更多的FLOPS,且无需依赖更低的精度或稀疏处理。

推理的瓶颈是内存带宽,而非计算?

事实上,对于像Llama-3这样的现代模型,并非如此!

使用英伟达和AMD的标准基准测试——2048个输入token和128个输出token,大多数AI产品的输入都要比输出长得多(即使是新的Claude聊天,系统提示也有1000多个token)。

在GPU和Sohu上,推理是以批次运行的。每个批次都会加载一次所有的模型权重,并在批次中的每个token上重复使用它们。

通常,LLM的输入是计算密集的,而LLM的输出是内存密集的。所以,当我们将输入和输出token与连续批处理结合时,工作负载就会变得非常「计算密集」。

下图的示例中,就展示了连续批处理LLM的过程。

这个过程中,会运行带有四个输入token和四个输出token的序列;每种颜色代表一个不同的序列。

同样的技巧,就可以扩展到2048个输入token和128个输出token的Llama-3-70B上。

要让每个batch,都包含一个序列的2048个输入token,和127个不同序列的127个输出token。

如果这样做的话,每个batch将需要大约(2048+127)×70B参数×每个参数2字节=304 TFLOPs,同时只需要加载70B参数×每个参数2字节=140 GB的模型权重,以及大约127× 64×8×128×(2048+127)×2×2=72GB的KV缓存权重。

这需要的计算,就远超过内存带宽的需求,因为一个H200需要6.8PFLOPS的计算能力,才能最大化其内存带宽。

而且,这是在100%利用率的情况下——如果利用率是30%,需要的计算能力还要多出3倍。

由于Sohu有极高的计算能力和高利用率,我们就可以在不受内存带宽限制的情况下,运行巨大的吞吐量。

注:在现实世界中,batch通常更大,输入长度各不相同,并且请求会以泊松分布到达。在这种情况下,Sohu的效果会更好。在这里之所以使用2048/128基准作为例子,是因为英伟达和AMD都在使用。

只需编写Transformer软件即可

无论在GPU和还是TPU上,软件都是一场噩梦。

处理任意的CUDA和PyTorch代码,需要的编译器极其复杂。

为此,AMD、英特尔、AWS这些第三方AI芯片,都在软件上投入了数十亿,但效果依然不佳。

这里,Sohu的好处就体现出来了——因为它只运行Transformer模型,所以我们只需要为Transformer模型编写软件!

大多数运行开源或内部模型的公司,都会使用特定的Transformer推理库,如TensorRT-LLM、vLLM或HuggingFace的TGI。

这些框架往往很死板,虽然我们可以调节模型的超参数,但并不支持更改底层的模型代码。

但是,没有关系!所有的Transformer模型都非常相似(即使是文本、图像、视频模型),调节超参数就足够了。

这样,就足以支持95%的AI公司了,不过,最大的AI实验室,仍然会进行定制化开发。

工程师团队会手动调优GPU内核,以挤出更多的利用率,并进行逆向工程,比如哪些寄存器对每个张量核心的延迟最低。

而创始人声称,有了Etched,我们就不再需要逆向工程了!

从驱动程序到内核再到服务框架,Etched的所有软件都会是开源的。

如果我们想实现一个自定义的Transformer层,内核专家完全可以自由地去做。

Etched已经破纪录,将成历史第一

现在看起来,Etced的决定很疯狂。

但更疯狂的是,他们是在2022年做出这项决定的——那时ChatGPT甚至还不存在!

当时,图像和视频生成模型还是U-Net模型,自动驾驶汽车是由卷积神经网络(CNNs)驱动的,Transformer架构还远未普及。

而现在,形势显然对他们非常有利。如今从语言到视觉,每个领域的顶尖模型都是Transformer。

这种趋同验证了Etced的前瞻性,更使Sohu成为十年来最重要的硬件项目。

种种迹象表明,公司正走在史上最快芯片发布的进程中——

- 顶尖的AI研究人员和硬件工程师纷纷离职原团队,加入Etced;

- Etced会直接和台积电合作开发4nm工艺,并且获得了足够的HBM和服务器,第一年的产量可以快速提升;

- Etced的早期客户,已经预订了数千万美元的硬件

「如果我们是对的,Sohu将改变世界」

如果AI模型在一夜之间,速度飙升20倍,且成本降低20倍,会发生什么?

目前,Gemini需要60多秒才能回答一个与视频相关的问题。

编码智能体的成本,比软件工程师更高,而且需要数小时才能完成任务。

视频模型每秒只能生成一帧,甚至ChatGPT注册用户达到1000万时(仅占全球用户的0.15%),就耗尽了OpenAI的GPU容量。

即便是持续以每两年2.5倍的速度增加GPU的容量,也得需要10年时间,才能实现「实时」视频生成。

而现在有了Sohu,这一切将瞬时完成。

网友表示,「AI的未来是定制硬件,实时视频模型即将现世」!

如果当实时视频、通话、AI智能体和搜索功能终于能够顺畅运行时,会发生什么呢?

很快,你就能知道了。

三位哈佛辍学生,挑战AI芯片霸主英伟达

成立于2022年,这家35人团队的初创公司Etched,坚信一定能够击败英伟达。

到底是什么样的背景,能让三位哈佛辍学生,敢于挑战芯片行业目前最炙手可热的赛道呢?

Robert Wachen、Gavin Uberti、Chris Zhu

创始人兼CEO Gavin Uberti自2020入学哈佛攻读数学专业,随后在2022年攻读硕士学位,专业是计算机。

不过,这两段上学经历,都以辍学告终。

在创办Etched之前,Gavin Uberti曾在另外两家公司分别有过一段全职和简直经历,担任了算法和后端工程师,以及软件工程师。

在进入大学之前,他参与了美国最著名的青少年科技创新大赛FIRST Tech Challenge,团队获得了Top 10奖项。团队开发的自动驾驶软件排在600个参赛团队第二名。

他还曾在2019年,拿过美式数学竞赛的冠军。

另一位创始人Chris Zhu,也是在哈佛就读时就在校外疯狂实习,甚至还没有从哈佛毕业,就已经成为兼职教员。

他个人的工作经历更为丰富些,不仅在高校MIT担任研究员、哈佛兼职教学研究员,还曾在亚马逊等公司做过实习。

最后一位联创兼COO Robert Wachen是一个多领域的创业者,曾创办了4家公司。

接下来,就看看他们会如何去缔造一个新的硅谷神话。

参考资料:

https://www.etched.com/announcing-etched

本文参与 腾讯云自媒体同步曝光计划,分享自微信公众号。
原始发表:2024-06-26,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 新智元 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 【新智元导读】史上最快Transformer芯片诞生了!用Sohu跑Llama 70B,推理性能已超B200十倍,超H100二十倍!刚刚,几位00后小哥从哈佛辍学后成立的公司Etached,宣布再融资1.2亿美元。
  • GPU撞墙了
  • 专用芯片,势在必行
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档