职位描述
岗位定位:
在性能架构师设定的基线目标与技术路线指导下,承担性能建模、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编程竞赛参赛经历或相关实验室研究经历者优先。