大模型分布式训练 Infra 实习生
招聘时间
即日起 至 2027-07-24
招聘单位
Momenta
所在地区
北京
工作地点
北京市/北京市/海淀区 学院8号写字楼5F
学历要求
硕士
实习时长
每周5天，每天3小时
职责描述
职位描述
1、参与 momenta 自动驾驶大模型、世界模型的分布式训练框架研发;
2、参与设计、研发、维护团队内部的模型分布式训练框架,擅长分析并深度优化训练各个阶段的性能瓶颈,包括计算效率、通信延迟、显存占用等;
3、结合不断迭代的模型算法逻辑,设计并实现针对性的高效分布式并行训练策略;
职位要求
1、学历背景:计算机相关专业,具备扎实的理论基础和强大的动手能力;
2、编程能力:熟悉Python/C++/CUDA/CUTLASS/Triton编程,熟悉PyTorch框架及其底层实现;
3、线下实习，长期实习6个月，每周实习4天及以上优先。
1、参与 momenta 自动驾驶大模型、世界模型的分布式训练框架研发;
2、参与设计、研发、维护团队内部的模型分布式训练框架,擅长分析并深度优化训练各个阶段的性能瓶颈,包括计算效率、通信延迟、显存占用等;
3、结合不断迭代的模型算法逻辑,设计并实现针对性的高效分布式并行训练策略;
职位要求
1、学历背景:计算机相关专业,具备扎实的理论基础和强大的动手能力;
2、编程能力:熟悉Python/C++/CUDA/CUTLASS/Triton编程,熟悉PyTorch框架及其底层实现;
3、线下实习，长期实习6个月，每周实习4天及以上优先。
