岗位描述
岗位描述：负责面向AMD GPU上的下一代大语言模型和视觉模型，进行深度学习算子的表示与Lowering开发，参与构建MLIR方言和算子Lowering能力，包括新的单一动态图方案。具体包括：参与单图动态图IR Lowering架构设计；实现新模型所需的注意力与控制流算子定义及MLIR Lowering；构建模型级和算子级运行状态上报能力并分析图优化器行为；参与MLIR方言规范及算子接入检查清单建设，评审算子Lowering工作；跟踪编译器、算法和架构趋势并应用到IR栈中；调试并修复现有问题，研究更高效的实现方式；与同事及合作伙伴建立技术协作关系。 岗位要求：热爱编译器和中间表示相关工作，具备推动复杂问题落地解决的能力，能够与AMD不同团队高效协作沟通。具备扎实的面向对象编程背景，优先C/C++，并注重细节；具备MLIR实践经验，包括方言、Lowering和Pass；深入理解深度学习算子和计算图；具备LLVM、GPU编程（HIP/CUDA）以及ONNX/ONNX Runtime经验者优先；熟悉注意力机制、Transformer与扩散模型算子内部原理，以及动态图和量化推理者优先；具备Windows和Linux开发经验，熟悉调试器、代码管理系统（GitHub）和性能分析工具；能够经常性使用AI编码智能体/助手提升日常工程效率；具备满足跨站点协作的英语工作能力，以及良好的沟通与问题解决能力。