高性能算子开发实习生

针对大模型训练、强化学习、推理场景，负责GPU kernel开发与调优

深入分析GPU/主流AI芯片的硬件架构特性，设计并实现高性能算子、算法和相关组件

保持关注行业前沿技术，且有能力和热情开展创新研究

在读硕士或博士，熟练掌握C/C++、Python编程语言，具备良好的coding和调试能力

熟悉GPU系统架构和CUDA/Cublas/Cutlass等，熟练使用nsys/nsight等分析工具

熟悉主流大模型训练/推理框架（Megatron-LM/vLLM/SGLang等），有实际性能调优经验 - 深入研究过至少一种机器学习框架（TF/PyTorch）或其他分布式训练框架（Megatron/VeRL等）的底层架构和原理 - 有硕士/博士阶段的计算机系统方向研究背景，或在国际顶级系统会议上（OSDI/SOSP/SIGCOMM/ASPLOS/NSDI等）发表过论文 - 有过完整参与LLM训练全过程的经历；有算法系统协同设计的经历