Ivyea Jobs 14720 roles · 250 companies · 刚刚 帮公司接 AI
芯元时代 · AI 芯片

高性能算子/Kernel工程师(2027届)

上海 / 深圳 / 北京 校招 · 正式 推理 / 部署训练系统 / 集群 工程开发

职位描述

岗位定位: 在性能架构师设定的基线目标与技术路线指导下,承担性能建模、Tiling(分块)实验与手写Kernel(内核)的开发工作。 工作职责 1. Roofline模型与性能建模: -针对GEMM(通用矩阵乘法)、Conv(卷积)、Attention(注意力)等算子,在给定硬件约束条件下评估算力与带宽瓶颈,输出系统性性能分析报告。 2. 手写实现与变体覆盖: -按照规格要求,实现指定Shape(形状)、Dtype(数据类型)及Layout(布局)的算子变体; -结合Tiling策略、数据复用及片上存储驻留进行对比实验,探索最优实现方案。 3. 精度验证与性能基准测试(Benchmark): -补齐精度对分验证、边界Shape测试及性能基准数据表格; -精度或性能指标未达标者,不得合入主分支。 4. 实现成果回流: -将经过验证的实现方案整理成可供代码生成(Codegen)参考的注释说明与最小化样例。

任职要求

任职要求: 1. 学历背景:统招全日制本科及以上学历,计算机、电子、自动化、软件工程、微电子等相关专业,211、985等重点学校优先。 2. 专业技能: -C/C++基础扎实,有良好的编码习惯; -理解并行计算基础知识(含SIMD、缓存层次、带宽与算力关系); -掌握GEMM、Conv、Attention等算子的计算模式与访存特征(具备课程或论文级理解即可); -了解CUDA、OpenCL或向量 intrinsic(内联函数)之一者更佳。 3. AI辅助开发意识:具备借助大语言模型(LLM)或智能体(Agent)工具提升开发效率的能力,同时具备对生成内容进行独立审查、调试与验证的能力和意识,以精度和性能指标约束产出质量。 加分项: 1. 具备CUDA Kernel实现经验、CUTLASS/Triton相关实验经验或FlashAttention类算法复现经验者优先; 2. 具备Roofline模型分析经验、Nsight/perf等性能分析工具使用经验者优先; 3. 了解量化算子(INT8 / INT4)优先; 4. 具备超级计算 / CUDA编程竞赛参赛经历或相关实验室研究经历者优先。
芯元时代去官方页面投递 →