东方算芯
*AI算子开发工程师
RESPONSIBILITIES
岗位职责
简体中文:我们正在寻找对底层技术充满热情、追求卓越性能的算子开发与优化工程师。您将深入探索AI芯片架构的极限,负责核心深度学习算子的设计、实现与性能调优,为AI模型提供算力基石。
岗位职责:
1. 核心算子研发与优化:基于自研架构特性,研发核心算子,支撑大规模训练与推理、性能建模与优化。
2. 底层指令级调优:基于汇编级指令调优,将硬件性能压榨至极限。
3. 定制CUTLASS/TileLang等高性能算子,基于这些库开发高性能算子(如 GEMM、FlashAttention)等。
4.跨层级性能攻坚,优化端到端效率,具备性能分析能力,熟练使用Profiler 等工具,能够精准定位并解决 Bank Conflict、内存带宽瓶颈、指令延迟等问题。
5. 深度学习基础:熟悉 Transformer、MoE 等主流模型架构,了解 vLLM、SGLang 等推理框架,能够理解模型计算特性并进行针对性的系统级优化。
6.适配类似Hopper/Blackwell等前沿架构。
REQUIREMENTS
任职要求
任职要求:
1.计算机等相关专业本科及以上;
2.精通C++模板元编程与CUDA;
3.熟悉汇编,能逆向优化指令,具备3年以上算子开发经验;
4.掌握CUTLASS等计算库具备算子实战开发经验;
5.熟练Profiler定位内存瓶颈;
6.熟悉Transformer/MoE等架构;
加分项:
有 DeepGEMM 或其他高性能GEMM库的开发或优化经验。
对FP8、FP4等低精度计算有深入理解和实践经验。
在相关会议上发表过相关论文,或为CUTLASS、PyTorch等主流开源项目贡献过核心代码。
地点:上海,北京,深圳,南京,西安,成都等地均可base
信息来自企业官方招聘渠道
OfferSeek 对公开岗位信息进行聚合、去重和结构化整理,最终申请以企业官方页面为准。