返回岗位列表

大模型优化工程师

查看原岗位

地点：北京薪资：30-60K 日期：2026-03-03

岗位摘要

负责PyTorch、Paddle等模型的适配与性能调优；完成多机多卡大模型训练部署及性能优化；负责vLLM、SGLang等推理引擎的部署与性能调优；开发高性能GPU算子，提升计算效率；对社区CUDA仓库进行musify适配

岗位职责

负责PyTorch、Paddle等模型的适配与性能调优
完成多机多卡大模型训练部署及性能优化
负责vLLM、SGLang等推理引擎的部署与性能调优
开发高性能GPU算子，提升计算效率
对社区CUDA仓库进行musify适配
调研最新机器学习技术与平台功能
深度参与产品新特性研发与支持

任职要求

积极乐观，责任心强，细致认真，具备良好团队协作与沟通能力
计算机科学、电子工程或相关专业本科及以上学历
熟悉机器学习与深度学习算法及其部署实践
至少掌握TensorFlow、PyTorch、Paddle、Megatron-LM、vLLM等一种框架的使用或开发
熟练使用CUDA、C++、Triton、Python中至少一种编程语言
具备扎实的数据结构、算法与逻辑思维能力

福利待遇

年薪30-60K×16薪，极具竞争力
参与前沿大模型与GPU性能极限探索
与顶尖AI工程化团队共同成长
北京望京国际研发园优质办公环境

工作地址

北京朝阳区望京国际研发园l座3层