大模型Infra工程师
岗位摘要
基于华为昇腾、平头哥等国产AI加速芯片，进行深度学习核心计算算子的高性能定制开发与迭代优化；负责将主流开源AI框架及大模型训练/推理工具链深度适配到国产算力平台，确保功能正确性与系统稳定性
岗位职责
- 基于华为昇腾、平头哥等国产AI加速芯片，进行深度学习核心计算算子的高性能定制开发与迭代优化
- 负责将主流开源AI框架及大模型训练/推理工具链深度适配到国产算力平台，确保功能正确性与系统稳定性
- 使用底层性能分析工具进行算子级与模型级的性能瓶颈分析，针对计算、访存、通信瓶颈制定调优策略
- 参与探索TVM、MLIR、Triton等AI编译器技术在国产异构后端的应用，提升算子自动生成效率与多硬件平台的泛化适配能力
任职要求
- 计算机科学、软件工程、电子工程、数学或相关专业本科及以上学历，具备扎实的计算机体系结构基础
- 深入理解GPU/NPU微架构、并行计算原理及内存一致性模型
- 精通C/C++与Python，具备出色的工程实现与架构设计能力
- 熟练掌握至少一种主流并行编程语言（CUDA/HIP/OpenCL/Ascend C），有丰富的底层算子开发、调试和调优经验
- 深入理解至少一种深度学习框架（PyTorch/TensorFlow/MindSpore）的底层运行机制
- 满足以下加分项之一者优先：具备华为昇腾CANN开发体系经验，熟练掌握Cube/Vector Core编程模型；有千亿级参数大模型在国产算力集群上的训练、微调或推理落地经验；在开源社区有核心代码贡献
加分项
- （满足其一即可优先考虑）
- 具备华为昇腾CANN开发体系经验，熟练掌握Cube/Vector Core编程模型者优先
- 有千亿级参数大模型（如LLaMA、Qwen、GLM）在国产算力集群上的训练、微调或推理落地经验者优先
- 在开源社区（如MindSpore、vLLM、FlashAttention、Cutlass等）有核心代码贡献者优先