职位描述
1.负责当前芯片架构的分析、评估与优化,针对计算、存储、互连、数据流等关键模块提出架构改进建议;
2.对软件编译器团队、算子团队提出的架构需求或修改方案进行分析,评估其合理性、收益、代价及对整体系统的影响;
3.跟踪并研究主流 AI 芯片架构,包括但不限于 GPU、TPU、昇腾、寒武纪等,阅读论文、白皮书及技术资料,完成竞品分析和架构对标;
4.构建并维护架构模拟器或性能模型,能够快速评估特定架构修改对性能、带宽、存储效率和系统瓶颈的影响;
5.研究 SOTA 大模型及其关键算子特征,分析当前芯片架构对模型的适配性,并提出针对性的架构优化方向;
6.与编译器、算子、设计实现等团队协作,推动架构分析结论落地。任职要求
1.计算机、电子工程、微电子、体系结构等相关专业,本科及以上学历;
2.具备扎实的计算机体系结构或AI芯片架构基础,对计算、存储、数据流、片上互连等有较深理解;
3.具备AI芯片、GPU/NPU/TPU、体系结构建模或性能分析相关经验;
4.能独立阅读并分析论文、白皮书及技术资料,具备较强的技术抽象和总结能力;
5.具备良好的建模与工程实现能力,熟悉Python、C++等至少一种开发语言,能够搭建分析工具或模拟器;
6.能理解主流大模型架构及关键算子特征,并完成模型需求到芯片架构的映射分析;
7.具备良好的逻辑分析能力、跨团队沟通能力和问题推动能力。
加分项:
1.有大模型训练或推理芯片架构经验;
2.有性能建模、模拟器开发或系统瓶颈分析经验;
3.熟悉Transformer、MoE、Attention、KVCache等模型和算子;
4.熟悉高带宽存储、3DIC、片上互连或数据流架构设计;
5.有GPU、TPU或其他主流AI芯片的深入分析经验。