职位描述
岗位职责:
1. 参与GPU平台相关软件开发,协助完成AI模型适配、GPU算子优化、软件与平台适配等工作,承担性能调优类任务;
2. 协助编写技术文档与培训材料,学习并支持GPU在数据中心、AI训练及推理等场景下的稳定运行;
3. 参与GPU产品在客户端的落地应用,协助排查应用场景中的问题及故障分析,实现GPU的广泛高效应用。任职要求
任职要求:
学历:研究生及以上,计算机、电子、图形学等相关专业应届毕业生;
基础:具备扎实的C/C++/Python编程基础,了解CUDA/ROCm等GPU并行计算语言者优先;有相关课程项目、实习或竞赛经历(如ACM、图形学/AI相关比赛)者优先;
技能:了解常见AI框架(如PyTorch、TensorFlow、Megatron-LM、DeepSpeed、vLLM、SGLang等)基本使用,对AI模型训练与推理流程有一定认知;具备良好的学习能力、逻辑思维与动手能力,能够快速上手GPU调试与性能分析工具;
素质:对GPU计算、AI基础设施方向有浓厚兴趣,具备较强的自驱力与团队协作精神,能适应快节奏的研发环境。