第一百九十八章 期待与你的合作

赵文渊回头看了看江松然的表情，内心暗爽。

天天看别人装逼，今天也让我装一回。

他表面上云淡风轻的说了一句：「是源码科技自己开发的，现在还在内测，估计很快要推向市场了。」

江松然半天没有说话，良久才又开口问道：「你们————咱们自己做的？」

赵文渊看江松然的心理承受能力还可以，就继续往下说：「是的，最早的灵感来源於源码的一款产品，开物。本来是面对没有技术背景的用户，後面是自研的代码大模型，用户完全不用懂代码，不用介入运行，用中文就可以进行叠代。」

赵文渊看了看江松然的反应，觉得他应该已经知道开物了。

「源码的姜总把这个思路应用到开发流程中来，做了这个智能体工具。」赵文渊说完，突然问，「你刚才在下面见过姜总了吗？」

江松然摇了摇头，他刚才直接和沈丛云上来了，没机会接触源码科技那边的员工。

但是他心里知道，能主导这个级别的项目，这个姜总一定是源码科技的核心人物，暗暗期待一睹他的风采。

「有机会介绍你认识，你会喜欢她的。」赵文渊说完，把话题拉回到工作上。

「总之，我们的工作流程就是这样的，把算子适配和叠代的工作分发到每个组员的手上，如果有人进度被卡住了再一起讨论解决的方案。」

江松然点了点头，怪不得他刚才看到的工作模式是那样的。

有这样的工具，就不奇怪了。

这麽看下来，这个工具大大的节省了具体实现代码、进行实际测试的时间，但它也不是什麽黑科技，更不是黑魔法。

能解决的问题，还是一样能解决，只是更快了。

不能解决的问题，靠这个工具依然解决不了。

对於真正懂原理的人来说，节省下来的时间最多，帮助也最大。

101看书

追书就上

101

看书网，s.?超实用

全手打无错站

反而对那些本来就一知半解的人来说，即使有了这样的工具用处也不大，解决不了的问题还是解决不了。

能做出十个算子的适配工作，这个技术团队的实力还是很强的。

他相信，凭着自己的技术能力和经验，配合着这套工具，加上这样的团队，把国产显卡的全链路适配做出来，不是没有可能。

江松然暗自点头。

然後他开口指着屏幕上的结果说：「这个算子暂时不用做到百分之八十。」

赵文渊一愣，问：「为什麽？」

「因为咱们现在缺的不是这个算子的单点性能。」江松然指了指屏幕，「它能跑，功能正确，性能有67%。如果目标是出论文，当然要把它调到80%以上，但如果目标是三月初跑通第一批国产卡集群，现在最应该做的不是继续死磕这个点。」

赵文渊的表情认真起来。

「那应该做什麽？」

「跑端到端。」

江松然把屏幕上的几个文件夹点开。

「把已经达标的这些算子拉出来，先组一条最小链路。模型可以小，分片可以小，数据可以用假的，但是链路必须完整，完成端到端的调试。

77

他一边说，一边在旁边的白板上写了起来。

模型加载、前向、反向、梯度同步、优化器更新、检查点备份、恢复。

江松然把七个步骤写在白板上，中间没有任何停滞。

他这几年做的全部事情都在这条链上，对这些知识、和实践中可能遇到的问题，全都已经烂熟於心了。

刚才写这个流程图，花时间最多的反而是在心里把这些词的英文都找到中文翻译。

流程图写完，江松然开口道：「虽然三月初只跑推理，但训练才是大头，端到端必须现在就摸清，不然到时候再补就来不及了。咱们现在用的这个「算子打靶法」是对的，但是既然时间紧、任务重，现在这个阶段最重要的是有大局观，从宏观上调节好优先级。」

他边说边用马克笔在白板上点了点：「否则，每个人都在优化自己手里的算子，等到最後拼起来的时候，真正的瓶颈才会显现出来，到那个时候就晚了。」

赵文渊看着白板上的内容，陷入了沉思。

江松然是对的。

迄今为止赵文渊带领团队做的都是基於「点」的工作，一个算子一个算子在推进。

这麽做当然是必要的，没有所有的必须的点，是不可能连成线的。

但是系统训练不只需要点，想要连线的时候不出错，那现在就得开始连了。

赵文渊算是见识到江松然的厉害了，入职的第一天上午，连项目进度都没看全，就把现在这个路线最大的隐患指了出来。

赵文渊转过椅子面

对江松然，说道：「江博士，你觉得我们现在应该怎麽办？」

江松然整理了一下思路，回答道：「第一阶段的目标如果只是跑推理的话，没必要追求N卡80%的效率，用不满，优先把必要的算子做完跑通，我刚才看了一下，除了十个已经标绿的，还有二十三个已经过了60%的线。」

江松然说到这，低头看了一下屏幕，自己纠正自己道：「二十四个，加起来是三十四个。」

江松然坐回椅子，把椅子往屏幕前挪了挪，开始在表格里圈算子。

「反向、优化器更新、梯度同步，这几类先排後，三月初的推理用不到，等四月调训练的时候再回来做。」

赵文渊点了点头，这几类在表里占了不少，确实都是训练才用得到的，先放一放完全合理。

「再看推理路径。」江松然把光标移到表格的前半段，「矩阵乘、LayerNorm

softma、激活，这几类已经标绿了。注意力相关的算子在哪儿？」

赵文渊往下翻了翻，指了一项：「标准Attention，61%。Flash—Attention那一版还没排上。」

「Flash不用，先用标准的，能跑就行。」江松然说，「KV—cache的读写呢？」

「这里，64%。」

「够了。」江松然往後靠了靠，「再加一个采样，一个RMSNorm。这两个简单，融合不融合都行，先有了再说，我数一下啊。」

他在表里点了点，停在最後。

「差两个。」

赵文渊愣了一下。

他原本打算的是把所有的算子都做到80%，卡住的地方直接呼叫随身老爷爷韩路一，按现在的速度可能要做到二月底了，到时候再做端到端。

但是现在被江松然这麽一拆—

只差两个了。

「你这————」赵文渊一时不知道怎麽接话了，「这两个我让组里今天就开。」

「不急。」江松然摆了摆手，「先把已经达标的这三十四个用现成的脚手架串起来，跑一遍空载，把链路打通，这两个等做好了再补进去，到时候哪儿不对一目了然。」

赵文渊点头，站起来转身去喊人。

他一边走一边又有点儿懊恼。

今天又装逼失败了。

一这是哪儿来的神仙啊。

赵文渊把手下的组员都叫过来，交代了任务，今天就开始搭脚手架，尝试用最小模型拉一个链路出来，在8卡环境下跑一圈。

周围的工程师都是一愣：「今天？」

江松然在旁边补了一句：「跑的时候把Profiler打开，不要只看最後的成败，每一步的耗时、等待时间、内存拷贝、通信比例，这些数据都有价值。」

众人还不认识江松然，都看向赵文渊，赵文渊点了点头，大家才散开。

赵文渊转过身来对江松然说：「江博士，谢谢你，这个确实是我没想到的点。还好你现在指出来了。」

江松然脸上倒是毫无骄傲的神情，他说：「这不是你的问题，咱们推进的太快了。」

赵文渊不解：「太快了？」

江松然说：「你知道我们当年做TPU的时候，花了多久才意识到这个问题吗？」

赵文渊递了个话：「多久？」

「三年。」江松然说，「你们现在才做了多久？最多半年吧？」

赵文渊已经没有纠正这个的欲望了，这点儿小逼不装也罢。

江松然接着说：「正常来说，一个团队还卡在第一个SDPA的时候，不需要想这麽远。

咱们现在的问题，是已经把最难启动的部分打穿了，所以系统性问题提前暴露了一」」

「这是好事。」

说完，江松然又打开赵文渊电脑上的那几个已经超过80%效率的算子文件夹，说道：「这些实现是你把关的吧？这得需要对底层有很深的理解才能做的出来，不管有没有智能体这种工具，这个理解是做不了假的。」

江松然比赵文渊年长几岁，手底下也管理过几十个人，他现在说这番话就是怕刚才的行为在下属面前伤了赵文渊的权威，所以想鼓励他一下，修补一下关系，防患於未然。

毕竟两人将来大概率要共事很久。

没想到赵文渊听完，脸上又出现了那种不自在的神色，纠结了一会儿，他才开口道：「其实，这些不是我写的。」

江松然好奇道：「那是谁写的？」

「是韩总写的。」赵文渊说着看了一眼电梯方向，最近韩路一上班之後都会直接来十三楼报导。今天听说是京城有人过来，国网投资的尽职调查团队，黄宇带队来的，需要韩路一去对接，所以他到现在还没来。

「韩总？」江松然今天惊讶太多次了，现

在已经有点儿惊讶不出来了，「哪些是他写的？」

赵文渊已经麻木了，你怎麽哪壶不开提哪壶呢。

「全是。」

「全是？」江松然强提一口气，又惊讶了，他回想了一下韩路一，那天在医院见到的年轻人。两人只见了一面，他当时还以为对方是HR。江松然本来完全不考虑再加入一个创业公司，但是韩路一说话就像有魔力一样，能直击他内心深处的需求，聊了短短几十分钟，他就被说服了。

从那之後，他觉得韩路一是个有天赋的管理者，他用他的技术来这里可以有所发挥，但是没想到，今天才知道，韩路一本人的技术也这麽强。

「韩总——」江松然犹豫了一下，八卦道，「是哪个学校的博士？」

「交大的本科。」赵文渊现在的心情很复杂，自己的逼没装成，但是替韩总装了一个，心里还莫名的有点儿爽。

这句话说完，两个人都陷入了沉默，然後默契的结束了话题。

江松然回到自己的新工位上，设置好了电脑，环顾了一下四周，还是乱糟糟的。

他突然想起了沈丛云说的，源智科技已经融资过了。好奇心起，他打开电脑，在飞书里大概搜索了一下，确实有相关的公告。

《源智科技完成4.5亿人民币A轮融资，加速国产算力基础设施研发》

4.5亿融资，估值30亿人民币。

江松然面露古怪。他本来以为自己是为了家庭拒绝了其他几家大厂的Offer，降薪入职源智科技。没想到除了200万的现金之外，那2%的期权现在就值六千万，这比鼎盛和崑仑这两家给的都多了一大截。

接下来江松然都有点心神不宁。

他不缺钱，但是这麽多钱，和这种信任，让他心中生出一种「士为知己者死」的感觉来。

终於等到韩路一给他发来飞书私信：「江博士，入职第一天还适应吗？不好意思我这边有别的事走不开，没能第一时间欢迎你加盟。有什麽问题可以谘询@赵文渊，他对这块最熟悉。」

江松然犹豫了一下，还是回复道：「韩总，我一切顺利，谢谢关心。关於期权奖励，实在丰厚，再次感谢。」

「期权在劳动合同上已经写明了吧，江博士对关於差旅的条款还有额外要求吗？」

江松然心想，是写了，但我没想到那不是纸钱啊。

他还没回复，韩路一又发了一条信息过来。

「我相信你能对得起这个价格，这不是工资，而是合伙人的证明。期待和你的合作。」

江松然沉默了一会儿，打出了自己的回覆。

「韩总，我也期待着和您的合作。」

这个话题告一段落，韩路一又开启了新的话题。

「刚才我听@赵文渊说了，你已经把训练链路大概理清楚了，这个工作量靠现有团队肯定不够，我这边会继续招人，但这个领域太窄，普通招聘效率太低。」

江松然明白了他的意思：「你想让我推荐些人？」

「你手底下的职位，如果你有合适的人，可以推过来见见。」

「好的。」江松然回复道。

然後他关掉飞书，打开了一个很久没说话的微信群。

【AIInfra交流群】

是个500人的大群，天南海北哪里的人都有，有美国的，也有国内的；有在Meta、

亚马逊这种大厂的，也有在创业公司的，还有一些在国内大厂的技术负责人。

唯一的共同点是，都是做这一行的。

江松然思考了一下，没有用源智科技的名义发信息源智科技现在声名不显，发出去反而让潜在的应聘者有顾虑。

他只以自己个人的名义发了一条信息。

「我回国了，现在在一个创业公司带Infra团队，方向是做国产AI加速卡的大模型训练栈。团队招人，方向包括训练系统、分布式调度、通信、容错、AIInfra工程化。有兴趣可以私信我。」

消息发出去没多久，许久没人说话的群活了过来。

「江神，你也回国了？」

「江神，你看好国产算力替代这个方向吗？」

「我前两天看了一篇论文，non—CUDAattention的，好像是一个国内团队写的？有人关注吗？」

「国产卡？训练？现在还有人幻想这个呢？」——看到这条，江松然嘴角微微上扬。

放在一个星期之前，他自己也是这个反应，但是他现在只觉得对方是井底之蛙。

「我有个师弟刚从新加坡回来，做调度了，要不要推给你？」

江松然知道自己在AIInfra这个领域的地位，这样一条信息发出去，过来发简历的人不会少的。

最近几天得花时间好好挑挑了。