阿里巴巴控股集团 · 校园招聘
高性能GPU算子生成 Agent 研发(AI for AI)-阿里星
杭州 ·经验不限·学历不限
北京·杭州
薪资面议
前往 阿里巴巴控股集团 官网投递
职位描述
1. 基于 LLM 的高性能算子/Kernel 代码生成系统开发:设计并开发异构GPU 高性能Kernel 的自动生成系统,覆盖算子实现、调度优化、内存管理、并行计算等关键模块;结合目标芯片架构特性(如计算单元、存储层次、指令集、通信机制等),持续提升生成代码的性能。
2. 模型后训练与 Agent 能力增强:构建面向高性能算子生成的数据闭环,设计并沉淀 SFT 数据集、偏好数据与 reward 机制;基于 SFT、RL 等后训练方法对模型进行微调,增强 Agent 对芯片约束、性能瓶颈和优化策略的理解,使其能够更稳定地生成高性能算子代码。
3. 多智能体协同系统构建:搭建多智能体协作框架,实现需求理解、代码生成、错误修复、版本对比等任务的自动化闭环;研究智能体分工、通信、反思与决策机制,提升系统在复杂场景下的鲁棒性、生成效率与优化质量。
4. 技术落地与前沿探索:与芯片、编译器和算子团队协作,结合具体硬件特性和业务场景推动技术落地;持续跟踪 LLM 代码生成、模型后训练、多智能体系统与 AI 编译优化等方向的前沿进展,并输出技术文档、专利或论文。
【任职要求】
1. 计算机科学、电子工程、人工智能或相关领域博士研究生。
2. 具备大语言模型及 Agent 系统开发经验,熟悉 Prompt Engineering、模型评测、RAG/向量检索及常见 Agent 框架;有代码生成相关项目经验者优先。
3. 熟悉模型后训练方法,具备 SFT、RLHF/RLAIF、DPO/PPO/GRPO 等一种或多种方法的实践经验,能够围绕业务目标构建训练数据、奖励机制与评测体系。
4. 熟悉 AI 芯片架构(如 GPU/NPU/TPU),掌握 CUDA、Triton、OpenCL、MLIR、TVM 等至少一种编程或优化技术。
5. 具备多智能体系统开发经验,熟悉任务规划、工具调用、反思优化、分布式协作等机制;有 AutoGen、LangGraph、MetaGPT 等框架经验者优先。
6. 扎实的编程能力,熟练使用 Python/C++,具备良好的系统设计与工程落地能力。
7. 具备编译原理、高性能计算(HPC)、算子开发、自动代码优化或 AI 编译器相关经验者优先。
8. 有训练数据构建、自动评测、模型迭代闭环建设经验者优先;有芯片软件栈、编译器或算子库实际落地经验者优先。
最后确认在招:2026-09-01 19:06:52 · 来源平台:alibaba