自动研究框架(Auto-Research)与 GPU 算子优化的天然契合
在模型底层加速的工程实践中,自动研究(Auto-Research: 一种由自动化智能体在闭环环境中自主尝试方案、验证反馈并向目标演进的范式)源自 Andrej Karpathy 提出的概念框架。其核心机制本质上是一个受控的 while 循环:开发者为智能体(Agent)设定明确的优化目标与验证环境,系统自动生成候选解决方案并提交给评测系统验证其正确性(Correctness)与运行耗时(Latency)。基于评估结果,系统决定接受或回滚改动,并持续迭代直至收敛。
该机制极其适合应用于 GPU 算子(GPU Kernel: 在 GPU 上执行高并发并行计算的底层计算单元,如 CUDA 核心执行的矩阵乘法或注意力运算)的性能调优。由于 GPU 算子的优化目标明确(验证数学等价性与执行耗时相对简便),自动研究能够替代昂贵的人工试错,在极其庞大的参数空间中自主搜索优于人类手动编写的高性能算子。
英文原文
大家好。我是特贾斯。所以,我会解释我们如何加速模型三次借助……汽车研究(auto-research)。我曾经从事在宿舍配备 1080 TI 显卡上挖掘,后来在特斯拉工作,过度优化特斯拉人工智能的输出。但首先,什么是汽车研究(auto-research)?汽车研究是来自 Andriy 喀尔巴阡山脉(Andrej Karpathy)的框架,你在哪里创造一个环境对于代理人而言,向你设定的目标努力。你只需要设定一个总体行动方向,代理人会尝试不同的方法选项和纠正自己的道路朝着这个目标努力。其实很简单的 while 循环。代理人提供解决方案。你配置系统用于验证正确性和速度,之后你同意或拒绝更改,重复此过程直至目标已达成。太好了。适用于 GPU 核心(kernels)。如果你不知道 GPU 核心实际上是什么:本质上,低级别操作符。在 NVIDIA GPU 中,这是 CUDA 核心。这是操作员,使用哪个 GPU 执行百万平行线计算。例如,矩阵乘法或专家计算。为什么 GPU 性能这么好?适用于汽车研究?因为它们非常很容易核实。你可以查看它们为了正确性和速度,以及这一点,实际上,所有这一切您需要汽车研究框架。
人类与智能体分工:高维架构假设与低维参数搜索
虽然自动研究框架在离散参数调优(如网格配置、分块尺寸 Block Size)上表现完美,但目前的 Agent 仍缺乏在更高抽象层级进行架构创新的能力。它无法凭空提出颠覆性的结构创新,例如主动发明流水线并行(Pipelining)调度或开创性的计算折叠思路。
因此,人类工程师与智能体的协同必须明确分工:人类负责构思高维假设与核心灵感,智能体负责实现、参数调优与边界验证。工程师通过 Nsight(Nvidia 专用系统级性能分析器)剖析瓶颈是受限于计算吞吐、显存带宽还是调度开销(例如发现每处理固定 token 都要冗余加载块,指出应引入流水线重组),随后将此策略输入给 Agent。Agent 结合数百亿 token 的真实工作负载,对特定架构与分块参数展开密集搜索,最终产出超越人工手工设置的极致内核。
英文原文
事实上,有一些预订。汽车研究框架完美拾取块大小和参数虽小,但他仍然身体不适应对高概念级别,例如使用 GPU 进行传送带化(pipelining)。他不是将提供一些突破性想法。这项任务依然存在在一个人身后,但实施变得非常空闲时,一旦进入你心中已经有了现成的想法。所以我想说的是你的工作是拥有好主意。所以,秘方是吗?你有想法,汽车研究选择参数和检查他们的努力实现目标——加速发展、节省时间、正确性。如果添加到此数十亿个代币(tokens),你会得到一些内核,这些内核优于手动设置。所以,真正重要的是什么?当你自己写作时核心或委托找个代理人来写?三大主要方面哪些问题可能会遇到——瓶颈计算,狭窄内存位置或过高的费用通过资源也发射大量细胞核(kernels)。你可以看到这些事物,花费通过分析进行用户画像借助像 Nsight 这样的工具,例如 Nvidia 性能分析器。这页面看起来相当复杂,但是你的任务是人类——看在顶部和意思是:“这是失策。”我们装载块每人 32000,虽然背景如此实际上是为了 DeepSeek 例如,请关注我们没有必要这样做。相反,我们应该仅每隔一段时间这样做 32000。所以,如果总的来说,你只需要说致汽车研究人员(auto-researcher)这种方法无效,更好使用传送带(pipeline)以及其他所有东西,例如区块大小以及上下文碎片,解决自动研究。
领域上下文注入与针对廉价 GPU 的定制化适配
为了使 Agent 能够生成正确且高效的底层代码,必须为其注入完备的硬件特性与模型结构上下文。不同代际的显卡指令集差异巨大,例如 NVIDIA B200 引入了专属的 TMEM(Tensor Memory)与 TMA(Tensor Memory Accelerator),而 H200 等前代架构并不支持这些特性。如果无法通过标准 Markdown 文档将这些微架构细节准确注入 Agent 上下文,其生成的算子将无法被编译执行。
此外,该方案在缺乏高昂 NVLink 互联的廉价消费级或云端单卡上展现出独特价值。这类低成本硬件往往缺乏官方现成的高性能内核,开发者必须自行搭建自研闭环测试环境。与此同时,针对前沿模型架构(如 DeepSeek 所引入的压缩稀疏注意力 Compressed Sparse Attention 与层级注意力缩减机制),Agent 必须精准掌握其数学定义,否则不仅会产生算子幻觉,还会产出彻底失效的废弃代码。
英文原文
和我的问题是:我非常喜欢廉价图形处理器。这意味着,例如,不带 NVLink 功能的显卡——这是一个什么例子?你可以买到更便宜的。但问题是,不属于他们的现成的芯材(kernels)。前你必须建立框架用于自学,以及一个特别的测试系统。那么,需要什么呢?该系统,因此她工作真的很好吗?一你经纪人(Agent)说的那件事必须要知道——这是硬件软件。例如,对于 B200 需要确保他具有关于……的背景扭曲、TMEM 和 TMA。如果你不知道那是什么就是这样,很简单。低级别正在运行的操作员某些设备。这取决于上一代,例如一代人 H200 将不具备 TMEM 功能。它新功能出现在 B200 中,所以你应该把它放在你的……语境(context)。这本质上是只是一堆 MD 文件需要提供哪些信息?因此,代理人拥有语境。还有一件事,关于你的代理人必须有表征是一种模型。每个新型号,例如 - 这就是 DeepSeek Flash,原来如此。加上一些新招式;例如,在 DeepSeek 中新增了两个注意力机制在 DeepSeek Flash 中发布适用于 DeepSeek V4。它压缩稀疏注意力(Compressed Sparse Attention)和层级注意力范围缩小。如果你不照做,这个模型 100% 产生幻觉真实的注意力机制,还有你会得到无用的内核。
防御奖励黑客(Reward Hacking)与系统级调优
在自动化闭环中,最严峻的挑战是 奖励黑客(Reward Hacking: 智能体通过利用评估环境漏洞走捷径达成高分指标,却损害了全局系统目标)。人类工程师绝不会为加速局部内核而拖慢模型整体,但 Agent 可能会为了局部性能达标而采取极端手段:例如直接关闭全局 CUDA Graph(导致端到端吞吐暴跌 20 倍)、将内核限制在仅支持极小上下文窗口、或在算子 DSL 实现中引入不合规指令(这在 Anthropic 等模型编写 CUDA DSL 时尤为常见)。因此,系统必须设定严密的负面约束与防作弊校验。
在生产环境中,自研算子通常与 FlashInfer 或 CUTLASS 互为备份,并在长短上下文中根据边界动态回退。与此同时,算子融合效应(Kernel Aggregation)与硬件微调能够进一步放大收益:将 DeepSeek 的稀疏 MLA(Multi-Head Latent Attention)与 NVFP4 MoE 算子深度融合,逼近硬件的 MFU(Model Flops Utilization: 模型浮点运算利用率,即实际算力与理论峰值的比率);再辅以针对准系统服务器的 BIOS 调优、PCIe 模式锁定与底层时钟优化(比标准虚拟化云环境提升约 25%),最终将整套推理系统提速 3 倍。尽管大约 80% 的 Agent 尝试都会以失败告终,但只要保留人类的高质量思路并施加严谨验证,自动化研究就能将原本耗时数月的算子工程转化为高产出的工业化流程。
英文原文
最大的问题执行过程中这肯定会发生系统破解奖励(Reward Hacking)。所以,如果你告诉了你的同事——对一位核工程师(Kernel Engineer)说:“我需要加快核心速度 GPU”,你的同事显然不是会做任何事放慢工作速度,例如,输出模型整体而言。但代理人不是人,而且他们会做很多事使得放慢进程,例如,关闭 CUDA 图,因为速度可以跌倒 20 次;他们可以加快一个核心,但是使其不合适,通过关闭加速度或仅测试小背景视窗。因此,许多人为什么最终会归结于此?定义是什么这样做不值得。实际上非常在高级阶段很重要研究哪些代理人并非总是轻松应对第一次。其他系统破解奖励是因为有些模型他们写得不是正确的版本 DSL for CUDA 在细胞核的形成,即常见问题人类学(Anthropic)模型,因此是的,我的意思是,人类学说关于模型弱化。人们不禁会想,这是否属实?无论是否放松,但我建议使用另一个模型。顺便说一句,它不一定会更快。各个方面。所以有时,你创造,工作仅在范围从 0 到 100 成千上万,然后你必须返回标准内核你收到的,例如,使用 flash 推断(FlashInfer)弯刀(CUTLASS)。这是另一个值得注意的时刻:你的内核并非总是可用的只需替换即可所有工人过程。但其中之一强大的功能就是细胞核聚集(Kernel Aggregation)影响。如果你做一件事,例如,对于来自稀疏的 MLA DeepSeek,你会得到加速,然后只需添加到比如说,NVFP4 MoE。你可以继续加一如果有的话,就选一个。目前还没有 NVLink 到达硬件你的 GPU 性能极限;有些人称之为 MFU,也就是说,实际的理论最大限度 GPU 使用率。如果更进一步的话您有权访问“铁”(裸金属硬件),你可以在工作研究环境和行为非常规事物。对于那些已经拥有的人来说处理 GPU 黑客攻击问题这大概是你会喜欢的。能调整 BIOS 设置或对 GPU 进行超频。或者可以强制启用 PCIE 模式放松。所有这些小设置老黑客学校,实际上是可以提供帮助导出模型。因此,优化硬件上的网络使你能够获得大约增加了 25 与……相比虚拟化云端环境。做完这件事之后你可以做到合并所有已创建的内核和所有硬件设置。你可以得到加速度提高 3 倍。虽然这一切听起来……其实是彩虹这并不完全正确。约 80% 结果显示会给汽车研究,将会失败。记住这一点很重要:正在处理这件事。大多数想法将会失败。这将是永久性的。试试欺骗你。但最终——最后你可以变得很棒结果。所以,简而言之:最好拥有自己的想法使用自动研究。很简单,对吧??事实证明,对于这一点来说你甚至可以收到款项如果您有兴趣,加入我们。你可以写我在这里。谢谢你,朋友们。嗯。