国产算力优化工程师

一、岗位职责
1.国产芯片底层软件栈适配
(1)作为公司与芯片厂商的核心技术接口人，深度对接沐曦、海光等国产算力卡商的底层软件团队。
(2)负责国产芯片驱动、编译器、运行时库及基础计算库的适配、升级与稳定性验证。
2.核心算子与性能问题优化
1)针对大模型核心算子进行定制化开发与极致优化，深入分析硬件架构特性，最大化释放国产芯片的算力潜能。
2)为公司AI领域的各类业务做算子层面的支持和适配，并推动在实际产品中的落地与应用。
3)与模型优化、适配、测试及产品开发团队积极沟通，基于客户实际场景与指标要求，提供技术支持并持续优化产品性能。
4)能够独立分析与定位性能问题的关键问题，并与算力厂商沟通，给出合理的建议方案。
3.异构计算环境调优
1)解决国产算力集群中的多卡通信、显存管理及分布式训练/推理的底层瓶颈问题。
2)配合适配、模型优化团队，解决其在调用底层算子时遇到的精度异常、性能回退及系统崩溃等疑难杂症。

二、任职要求
（一）必备条件
1)学历背景：计算机、人工智能等相关专业本科及以上学历。
2)工作经验：3年以上高性能计算、AI芯片底层软件开发或编译器开发经验。
3)编程与底层能力：
a.精通 C/C++，熟悉 Python，具备极强的系统级代码调试能力。
b.深入理解计算机体系结构、并行计算原理及 GPU/NPU 硬件架构。
c.熟练运用CUDA等编程原语开发AI领域的各类算子；熟悉沐曦、海光等国产算力芯片生态，具备至少一代国产AI芯片架构的算子移植、开发与优化经验。
d.熟悉并行计算技术，包括但不限于内存访问模式优化、资源利用率优化、计算通信重叠等。熟悉关键性能指标的含义及影响因素，掌握芯片厂商性能分析工具的使用，能独立完成指令级与算子级的性能分析。
4)算子与编译器经验：
a.熟悉 LLVM 等编译器框架的实战经验。
b.能熟练使用工具进行指令级性能分析。
c.熟悉现代大算力AI芯片体系结构，包括但不限于并行调度模型、存储层次结构、计算单元架构等。

（二）加分项
1)有SDD、TDD经验，熟悉使用Trae、CodeBuddy等国内智能体优先。
2)有社区贡献经验优先。