大模型Infra算法实习生
岗位摘要
超大规模GPU集群分布式训练技术的研究与实现,提升MOE大模型和强化学习的训练吞吐;大模型前沿算法的调研和实现;大模型训练框架和推理框架的研发与性能调优;熟悉Python或C++,具备扎实的算法和计算机基础、良好的编程风格
登录后保留你的岗位线索
邮箱登录后可继续查看完整职责、任职要求和原岗位入口,并使用收藏与浏览记录沉淀适合自己的机会。
岗位职责
- 超大规模GPU集群分布式训练技术的研究与实现,提升MOE大模型和强化学习的训练吞吐
- 大模型前沿算法的调研和实现
- 大模型训练框架和推理框架的研发与性能调优
任职要求
- 熟悉Python或C++,具备扎实的算法和计算机基础、良好的编程风格
- 熟悉深度学习和大模型训练相关算法
- 熟练使用深度学习框架(包括PyTorch/JAX/Paddle等)
- 有强烈的工作责任心,良好的学习能力、沟通能力和自驱力
- 有大规模GPU集群性能优化和训练经验者优先
- 熟悉大模型训练框架和推理框架(如Megatron/DeepSpeed/ColossalAI/torchtitan/sglang/vllm等)者优先
- 熟悉大模型算法(如大模型基础架构/强化学习/多模态/视频生成等)者优先
- 在顶级会议发表论文或有竞赛经验者优先
- 熟悉CUDA相关开发者优先
加分项
- 有大规模GPU集群性能优化和训练经验优先
- 熟悉大模型训练框架和推理框架,包括不限于Megatron/DeepSpeed/ColossalAI/torchtitan/sglang/vllm等
- 熟悉大模型算法,如大模型基础架构/强化学习/多模态/视频生成等
- 在顶级会议发表论文或有竞赛经验优先考虑
- 熟悉cuda相关开发
福利待遇
- 日薪200-300元,薪酬具有竞争力
- 参与千卡以上超大规模GPU集群训练项目
- 深入接触大模型前沿算法与Infra技术
- 表现优异者有转正机会
工作地址
北京朝阳区望京国际研发园l座3层
登录后查看完整岗位详情
使用邮箱验证码登录后,可查看完整职责、任职要求、原岗位入口,并继续使用收藏和浏览记录。列表摘要保持公开,完整详情用于保护数据质量和降低恶意抓取。