Ivyea Jobs 14660 roles · 250 companies · 刚刚 帮公司接 AI
东方算芯 · AI for Science

*AI算子开发工程师

上海 社招 · 全职 社会招聘 推理 / 部署 工程开发

职位描述

简体中文:我们正在寻找对底层技术充满热情、追求卓越性能的算子开发与优化工程师。您将深入探索AI芯片架构的极限,负责核心深度学习算子的设计、实现与性能调优,为AI模型提供算力基石。 岗位职责: 1. 核心算子研发与优化:基于自研架构特性,研发核心算子,支撑大规模训练与推理、性能建模与优化。 2. 底层指令级调优:基于汇编级指令调优,将硬件性能压榨至极限。 3. 定制CUTLASS/TileLang等高性能算子,基于这些库开发高性能算子(如 GEMM、FlashAttention)等。 4.跨层级性能攻坚,优化端到端效率,具备性能分析能力,熟练使用Profiler 等工具,能够精准定位并解决 Bank Conflict、内存带宽瓶颈、指令延迟等问题。 5. 深度学习基础:熟悉 Transformer、MoE 等主流模型架构,了解 vLLM、SGLang 等推理框架,能够理解模型计算特性并进行针对性的系统级优化。 6.适配类似Hopper/Blackwell等前沿架构。

任职要求

任职要求: 1.计算机等相关专业本科及以上; 2.精通C++模板元编程与CUDA; 3.熟悉汇编,能逆向优化指令,具备3年以上算子开发经验; 4.掌握CUTLASS等计算库具备算子实战开发经验; 5.熟练Profiler定位内存瓶颈; 6.熟悉Transformer/MoE等架构; 加分项: 有 DeepGEMM 或其他高性能GEMM库的开发或优化经验。 对FP8、FP4等低精度计算有深入理解和实践经验。 在相关会议上发表过相关论文,或为CUTLASS、PyTorch等主流开源项目贡献过核心代码。 地点:上海,北京,深圳,南京,西安,成都等地均可base
东方算芯去官方页面投递 →