曙光信息产业股份有限公司
【2027】研发工程师(AI方向)-郑州(J14226)
RESPONSIBILITIES
岗位职责
岗位职责
参与 GEMM、Attention、MoE、量化/反量化等核心算子的性能分析与优化。
分析 AllReduce、AllGather、All-to-All 等分布式通信开销。
参与 kernel fusion、算子调度、编译优化和硬件特性适配。
使用 profiler 工具定位 GPU kernel、显存访问、通信链路等性能瓶颈。
配合框架和部署团队完成端到端性能优化验证。
REQUIREMENTS
任职要求
任职资格
计算机、电子信息、自动化、数学等相关专业,硕士优先。
熟悉 C/C++ 或 Python,具备扎实的数据结构、操作系统和计算机体系结构基础。
了解 CUDA、GPU 编程、并行计算或高性能计算基础。
理解矩阵计算、Attention、MoE、量化等大模型核心计算模式。
对底层性能优化有强兴趣,具备较好的代码阅读和问题钻研能力。
加分项
有 CUDA、Triton、TVM、MLIR、NCCL、CUTLASS 等实践经验。
参加过 HPC、编译器、并行计算、深度学习系统相关项目。
熟悉 Nsight、CUPTI、torch profiler 等性能分析工具。
信息来自企业官方招聘渠道
OfferSeek 对公开岗位信息进行聚合、去重和结构化整理,最终申请以企业官方页面为准。