大模型训推优化工程师-算子优化方向
岗位摘要
负责大模型训练与推理过程中算子的性能分析与优化；设计并实现高效算子，提升模型计算效率与硬件利用率；与框架及硬件团队协同，推进算子在实际场景中的落地与迭代；本科及以上学历，计算机/电子/信息/通信/自动化/软件等相关专业优先
岗位职责
- 负责大模型训练与推理过程中算子的性能分析与优化
- 设计并实现高效算子，提升模型计算效率与硬件利用率
- 与框架及硬件团队协同，推进算子在实际场景中的落地与迭代
任职要求
- 本科及以上学历，计算机/电子/信息/通信/自动化/软件等相关专业优先
- 深入理解计算机体系结构，掌握C++等语言
- 熟悉CUDA/AscendC/BangC等的一种或者多种
- 具备良好的交流沟通能力、业务意识、团队合作精神
工作地址
北京海淀区大恒科技大厦南座12层