大模型训练最可怕的地方就在这里:不是做到“能跑”就算成功,还必须稳定、高效、鲁棒。
必须在极端昂贵的算力成本下,把每一张卡的利用率挤压到极致。
否则你花了同样的钱,别人训练只要一个月,你训练却要三个月。
或者同样的训练量,别人只需要一千万,你却要烧掉三千万。
最后模型效果还不一定赶得上。
赵文渊潜心研究了两天,越研究越觉得难做。
他发现最大的问题不是技术上完全不可行;恰恰相反,很多东西理论上都有办法做。
但工程量太大了。
这可不是一个天才程序员闭关三个月就能解决的问题。
他面对的,是一个行业十多年、无数公司、无数开发者、无数论文、无数开源项目共同堆起来的生态。
这就像愚公移山:别说赵文渊一个人,别说源码科技模型组这点人,就算韩路一给他一个上百人的开发团队、花几年时间,可能也只够把最核心的部分做出一个可用版本。
这现实吗?
不现实。
赵文渊狠狠抓了抓头发,看看抓掉的头发,又心疼地摸了摸。
他打开微信,找到一位在谷歌工作时的华人同事。
那位同事当年给tpu做过适配。谷歌自研芯片加自研框架的路线面临同样困境,只是他们当年砸钱砸出来了。
赵文渊写了一条长长的信息,把自己的困境简短说了说。
如何从训练框架层面对新硬件做适配?自研芯片生态早期最难补的短板是什么?如果一个创业团队想在国产ai芯片上跑大模型,有没有现实些的切入方式?
这里面可能涉及保密内容,赵文渊和这位前同事也两年没联系了,本不预期一定收到回复。
没想到半个小时之后,对方回复了。
看到内容,赵文渊气笑了。