机器学习平台开发工程师

查看原岗位

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

岗位摘要

负责PyTorch模型适配与性能调优；负责多机多卡分布式大模型训练部署与性能调优；负责vLLM及推理引擎部署与性能调优；开发高性能算子；对社区CUDA仓库进行musify；调研最新机器学习技术

积极乐观，责任心强，工作认真细致，具有良好的团队沟通与协作能力
计算机科学、电子工程或相关领域本科及以上学历
熟悉机器学习与深度学习算法及实际部署
至少熟悉一种机器学习框架的使用或开发，如TensorFlow、PyTorch、PaddlePaddle、Megatron-LM、Colossal-ai、DeepSpeed、vLLM或推理引擎
熟悉至少一种编程语言：CUDA、C++、Python
具备良好的逻辑思维能力，精通计算机数据结构与算法
有2年及以上技术支持或相关领域经验者优先
熟悉CUDA编程者优先
熟悉Docker、Kubernetes等容器化技术者优先

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