异构适配与优化AI Agent研发工程师/专家（深圳/北京/上海）

岗位职责

1. 构建算子开发 Agent，基于 LLM 自动生成后端算子，覆盖代码生成、正确性校验、Benchmark 回归的完整闭环；
2. 构建跨芯迁移 Agent，将现有 GPU 算子与框架适配代码自动移植到相关芯片后端；
3. 负责 vLLM / SGLang 等框架在相关芯片上的适配落地，并对 MoE、Attention、KV Cache 等大模型关键路径做算子级深度优化；
4. 性能分析与调优，Profiling 定位瓶颈，在计算、显存、通信三个维度持续榨干硬件性能；
5. 沉淀专家经验为 Agent 可用的知识、规则与工具，实现自动调优闭环。

任职要求

1. 具备 LLM Agent 开发实践（Prompt Engineering、Tool Use、代码生成与校验），有 Coding Agent 深度使用或二次开发经验；
2. 有系统性性能优化方法论与真实提速案例，工程能力扎实，熟悉 CI/CD 与自动化测试；
3. 有AI Infra / 高性能计算经验，精通 CUDA 或 Triton，有手写高性能算子的实战经历；
4. 熟悉 PyTorch 算子机制与 vLLM / SGLang 等主流推理框架，了解大模型推理关键机制（KV、PD分离等）；
5. 有相关芯片适配经验、AI 算子生成项目经验者优先。

产品/项目

大数据(数据计算平台部)

工作年限

三年以上工作经验