高性能算子开发工程师

岗位职责

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

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

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

任职要求

1. 熟练掌握C/C++、Python编程语言，具备良好的coding和调试能力

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

3. 熟悉主流大模型训练/推理框架（Megatron-LM/vLLM/SGLang等），有实际性能调优经验