AI编译器工程师

【岗位职责】参与AI编译器前端/调度/后端的设计与开发，实现芯片高性能算子库。参与AI模型的图优化、图拆分、算子融合、调度优化、指令生成等。协同芯片、框架、驱动等团队，分析和优化编译器性能。编写高质量、可测试、可维护的代码，并输出高质量文档。负责大模型在推理侧的性能优化与加速工作，聚焦模型量化、投机采样、KV Cache管理等核心方向。推动大模型在公司自研芯片上的高效部署，包括计算图导出、图优化、算子融合及性能调优。持续跟进大模型推理领域的前沿技术，快速进行方案验证与工程落地。与编译团队协作，共同推进大模型在自研芯片的高效运行。

【任职要求】计算机、电子工程、自动化、人工智能等相关专业，研究生及以上学历。熟悉编译原理，掌握数据流图、中间表示等基础知识。熟悉LLM与CNN模型，了解常见算子的详细计算方法。熟悉C++/Python编程，具备良好的编码能力。有LLVM、MLIR、TVM、TensorRT、XLA等相关编译器项目经验者优先。有AI框架（如PyTorch、TensorFlow、ONNX）使用或对接经验者加分。具备良好的学习能力、沟通能力和团队协作意识。硕士及以上学历，计算机科学及相关专业。熟悉Transformer等主流大模型架构，理解其推理流程，具备扎实的深度学习基础。熟练掌握PyTorch等主流深度学习框架，

今日招聘宣讲会 20260612  10:00  职业发展中心美团创新空间 中国铁设2027届应届毕业生招聘