
米哈游|高性能算子开发实习生
岗位职责
1.针对大模型训练、强化学习、推理场景,负责GPU kernel开发与调优
2.深入分析GPU/主流AI芯片的硬件架构特性,设计并实现高性能算子、算法和相关组件
3.保持关注行业前沿技术,且有能力和热情开展创新研究
任职要求
1.在读硕士或博士,熟练掌握C/C++、Python编程语言,具备良好的coding和调试能力
2.熟悉GPU系统架构和CUDA/Cublas/Cutlass等,熟练使用nsys/nsight等分析工具
3.熟悉主流大模型训练/推理框架(Megatron-LM/vLLM/SGLang等),有实际性能调优经验
加分项
- 深入研究过至少一种机器学习框架(TF/PyTorch)或其他分布式训练框架(Megatron/VeRL等)的底层架构和原理
- 有硕士/博士阶段的计算机系统方向研究背景,或在国际顶级系统会议上(OSDI/SOSP/SIGCOMM/ASPLOS/NSDI等)发表过论文
- 有过完整参与LLM训练全过程的经历;有算法系统协同设计的经历
米哈游招聘实习生:高性能算子开发实习生岗位来自米哈游招聘官网,具体内容以米哈游招聘官网为准。
