岗位描述
职位描述:
1)负责 AI 编译器的设计、开发与优化,提升编译器性能,降低计算资源消耗,提高模型执行效率;
2)参与 AI 编译器核心模块的架构设计与代码实现,解决编译器在实际应用中的技术难题;
3)与 AI 算法团队、硬件团队紧密合作,针对不同的硬件平台和 AI 模型进行编译器的适配和优化;
4)跟踪 AI 编译器领域的最新技术发展趋势,引入新技术、新方法,推动 AI 编译器的技术创新和产品升级;
5)对 AI 编译器进行性能测试和分析,提出性能改进方案,并负责方案的实施和验证。
任职要求:
1)计算机科学、电子工程、数学等相关专业,硕士及以上学历;
2)精通 C/C++、Python 等编程语言,具备扎实的编程基础和良好的代码风格;
3)深入理解编译器原理,熟悉编译流程(词法分析、语法分析、语义分析、中间代码生成、代码优化、目标代码生成等),掌握至少一种主流编译器框架(如 Triton、TVM、MLIR等)的使用和开发;
4)熟悉深度学习框架(如 TensorFlow、PyTorch 等)的原理和使用,了解深度学习模型的训练和推理过程;
5)了解硬件架构(如 CPU、GPU、DSA 等)的基本原理,能够针对不同硬件平台进行性能优化;
6)具有编译器开发或相关领域的工作经验;
7)参与过大型编译器项目或在开源编译器项目中做出贡献的开发者优先考虑。
其他要求:
1)具备较强的学习能力和创新能力,能够快速掌握新知识、新技术;
2)具有良好的团队协作精神和沟通能力,能够与不同背景的人员有效合作;
3)具备较强的问题解决能力和抗压能力,能够在复杂的工作环境中独立完成任务。