大模型训练推理优化工程师

查看原岗位

地点：北京薪资：25-50K 日期：2026-03-03

岗位摘要

负责PyTorch模型适配与性能调优；负责多机多卡分布式大模型训练部署与性能优化；负责vLLM/SGLang等推理引擎部署与性能优化；开发高性能CUDA算子；对社区CUDA仓库进行musify优化

积极乐观，责任心强，工作细致，具备良好团队协作与沟通能力
计算机科学、电子工程或相关专业本科及以上学历
熟悉机器学习与深度学习算法及实际部署
至少熟悉一种主流机器学习框架（TensorFlow、PyTorch、PaddlePaddle、Megatron-LM、Colossal-AI、DeepSpeed、vLLM等）
至少精通一种编程语言：CUDA、C++或Python
具备扎实的数据结构与算法基础，逻辑清晰
2年以上相关技术支持经验者优先
熟悉CUDA编程者优先
熟悉Docker、Kubernetes等容器化技术者优先

北京朝阳区望京国际研发园I座