返回岗位列表

高性能计算工程师

查看原岗位

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

岗位摘要

研究并实现超大规模GPU集群分布式训练技术，提升MOE大模型与强化学习训练吞吐；调研并实现前沿大模型算法；研发并优化大模型训练与推理框架；熟悉Python或C++，算法与计算机基础扎实，编程风格良好

岗位职责

研究并实现超大规模GPU集群分布式训练技术，提升MOE大模型与强化学习训练吞吐
调研并实现前沿大模型算法
研发并优化大模型训练与推理框架

任职要求

熟悉Python或C++，算法与计算机基础扎实，编程风格良好
熟悉深度学习及大模型训练相关算法
熟练使用PyTorch/JAX/Paddle等深度学习框架
责任心强，学习、沟通与自驱力佳
有大规模GPU集群性能优化及训练经验优先
熟悉Megatron/DeepSpeed等训练/推理框架优先
熟悉大模型算法如强化学习、多模态、视频生成优先
顶会论文或竞赛经验优先
熟悉CUDA开发优先

加分项

有大规模GPU集群性能优化和训练经验优先
熟悉大模型训练框架和推理框架，包括不限于Megatron/DeepSpeed/ColossalAI/torchtitan/sglang/vllm等
熟悉大模型算法，如大模型基础架构/强化学习/多模态/视频生成等
在顶级会议发表论文或有竞赛经验优先考虑
熟悉cuda相关开发
胡先生刚刚活跃

福利待遇

月薪40-60K，16薪
北京望京国际研发园办公，环境优越
与顶尖AI团队共事，技术成长快

工作地址

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